QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#488039#6638. Treelectionucup-team052#WA 589ms52132kbC++231.4kb2024-07-23 15:30:002024-07-23 15:30:00

Judging History

你现在查看的是最新测评结果

  • [2024-07-23 15:30:00]
  • 评测
  • 测评结果:WA
  • 用时:589ms
  • 内存:52132kb
  • [2024-07-23 15:30:00]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 5;

vector <int> adj[N];
int fa[N], siz[N], ans[N];
int T, n;

void dfs1(int u) {
	siz[u] = 1;
	for (auto v : adj[u]) {
		dfs1(v);
		siz[u] += siz[v];
	}
}

int f[N], MID;
void dfs2(int u) {
	f[u] = 0;
	for (auto v : adj[u]) {
		dfs2(v);
		f[u] += f[v];
	}
	f[u] = max(0, f[u] - MID) + (u != 1);
}

int ok[N];
void dfs3(int u) {
	if (f[u] <= 1) return;
	ok[u] = 1;
	for (auto v : adj[u]) dfs3(v);
}

int main() {
	scanf("%d", &T);
	while (T--) {
		scanf("%d", &n);
		for (int i = 1; i <= n; i++) {
			adj[i].clear();
			ok[i] = 0;
		}
		for (int i = 2; i <= n; i++) {
			scanf("%d", &fa[i]);
			adj[fa[i]].push_back(i);
		}
		if (n == 2) {
			printf("10\n");
			continue;
		}
		dfs1(1);
		int l = 1, r = n - 1, res = n - 1;
		while (l <= r) {
			int mid = (l + r) >> 1;
			MID = mid; dfs2(1);
			if (f[1] <= 1) res = mid, r = mid - 1;
			else l = mid + 1;
		}
		MID = res; dfs2(1);
		if (f[1] == 1) {
			for (auto v : adj[1]) dfs3(v);
		} else {
			for (int i = 1; i <= n; i++) ok[i] = 1;
		}
		// fprintf(stderr, "res = %d, f[1] = %d, f[2] = %d\n", res, f[1], f[2]);
		for (int u = 1; u <= n; u++) {
			int ans = 0;
			if (siz[u] - 2 > res) ans = 1;
			else if (siz[u] - 2 == res) ans = ok[u];
			printf("%d", ans);
		}
		printf("\n");
	}
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 3ms
memory: 12088kb

input:

2
4
1 2 3
5
1 1 2 2

output:

1100
10000

result:

ok 2 lines

Test #2:

score: 0
Accepted
time: 182ms
memory: 19008kb

input:

10
100000
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 10...

output:

111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111...

result:

ok 10 lines

Test #3:

score: 0
Accepted
time: 233ms
memory: 52132kb

input:

1
1000000
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 10...

output:

111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111...

result:

ok single line: '111111111111111111111111111111...0000000000000000000000000000000'

Test #4:

score: 0
Accepted
time: 589ms
memory: 41736kb

input:

1
1000000
1 2 2 3 5 1 7 3 9 8 4 11 9 6 10 12 10 13 16 18 14 16 7 24 23 21 14 8 26 27 17 21 33 24 20 17 5 34 29 34 37 20 12 22 39 44 33 37 32 27 31 25 30 46 39 26 53 4 11 55 41 29 6 22 32 36 52 57 65 38 59 67 51 56 30 61 36 64 62 19 51 63 53 83 38 75 31 41 49 43 60 18 62 67 91 81 44 25 55 98 86 64 46...

output:

111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111...

result:

ok single line: '111111111111111111111111111111...0000000000000000000000000000000'

Test #5:

score: -100
Wrong Answer
time: 173ms
memory: 18384kb

input:

65764
13
1 1 1 1 1 1 1 1 2 10 10 11
11
1 1 2 2 2 4 4 4 5 10
14
1 2 2 3 3 3 4 4 5 6 10 12 13
14
1 2 2 3 3 3 4 5 5 6 6 7 8
13
1 2 2 3 3 4 5 5 5 6 8 12
13
1 1 1 2 2 3 5 7 7 7 9 9
12
1 1 1 2 2 3 5 7 7 9 10
14
1 1 2 2 3 4 5 6 7 7 8 12 12
14
1 2 3 4 4 5 5 5 7 7 7 10 13
14
1 1 1 1 1 2 2 2 3 10 10 10 11
13
...

output:

1000000000000
11000000000
11101000010000
11100000000000
1110100000000
1010001000000
101000100000
11011001000000
11111010000000
10000000000000
1111010000000
111000100010000
111010001000
11000000000000
11100000000000
11001001000000
11000000000000
11111000000000
110011000000
1010000100000
1100111010000...

result:

wrong answer 8604th lines differ - expected: '10000000000', found: '00000000000'