QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#113554#6638. TreelectionAvavaAvaWA 937ms94244kbC++141.4kb2023-06-18 15:34:312023-06-18 15:34:34

Judging History

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

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-06-18 15:34:34]
  • 评测
  • 测评结果:WA
  • 用时:937ms
  • 内存:94244kb
  • [2023-06-18 15:34:31]
  • 提交

answer

#include <bits/stdc++.h>
#define int long long
#define mod 1000000007
using namespace std;
vector<int> e[1000005];
int sz[1000005],fa[1000005],f[1000005],g[1000005],ok[1000005];
int k,n;
void dfs2(int u)
{
	int sum=0;
	for(auto v:e[u]) sum+=f[v]-min(f[v],k)+1;
	for(auto v:e[u])
	{
		if((sum-(f[v]-min(f[v],k)))>g[u]) g[v]=-1;
		else g[v]=k-(sum-(f[v]-min(f[v],k)))+k;
	}
	for(auto v:e[u]) dfs2(v);
}
void dfs1(int u)
{
	f[u]=0;
	for(auto v:e[u]) dfs1(v),f[u]+=f[v]-min(f[v],k)+1;
}
bool check(int S)
{
	k=S-2;
	if(k<0) return 0;
	dfs1(1),g[1]=k,dfs2(1);
	int flag=0;
	for(int i=1;i<=n;i++)
	{
		if(g[i]>=0&&sz[i]==S)
		{
			ok[i]=1;
			flag=1;
		}
	}
	return flag;
}
void solve()
{
	for(int i=1;i<=n;i++) sz[i]=0,e[i].clear(),ok[i]=0;
	cin >> n;
	vector<int> S;
	for(int i=2;i<=n;i++)
	{
		cin >> fa[i];
		e[fa[i]].push_back(i);
	}
	for(int i=n;i>=1;i--) sz[fa[i]]+=++sz[i],S.push_back(sz[i]);
	sort(S.begin(),S.end());
	int l=0,r=n-1,p=n;
	while(l<=r)
	{
		int mid=(l+r)/2;
		if(check(S[mid])) p=S[mid],r=mid-1;
		else l=mid+1;
	}
	for(int i=1;i<=n;i++) if(sz[i]>p) ok[i]=1;
	for(int i=1;i<=n;i++) cout << ok[i];
	cout << "\n";
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int T;
	cin >> T;
	while(T--) solve();
	return 0;
}
/*
2
7
1 2 2 1 5 5
7
1 2 2 1 5 6


5
4
1 2 3
5
1 1 2 2
8
1 2 3 4 5 6 7
7
1 2 2 1 4 5
7
1 2 2 1 5 6
*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 33308kb

input:

2
4
1 2 3
5
1 1 2 2

output:

1100
10000

result:

ok 2 lines

Test #2:

score: 0
Accepted
time: 164ms
memory: 41540kb

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: 197ms
memory: 94244kb

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: 937ms
memory: 86516kb

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: 229ms
memory: 38172kb

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
11101000000000
11100000000000
1110100000000
1010001000000
101000100000
11011001000000
11111010000000
10000000000000
1111000000000
111000100010000
111010001000
11000000000000
11100000000000
11001001000000
11000000000000
11111000000000
110011000000
1010000100000
1100111010000...

result:

wrong answer 3rd lines differ - expected: '11101000010000', found: '11101000000000'