Submission #107536

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
107536	2019-04-25T01:49:58 Z	aer0park	None (KOI16_tree)	C++14	36 / 100	315 ms	27896 KB

tree

#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
 
ll n,q,cnt=1,qy=1;
ll ind[200004],pw[200004],chd[200004],par[200004];
vector<ll> g[200004];  //원래 번호 
 
ll val(ll a)
{
	ll ret=0;
	while(a)
	{
		ret+=pw[a];
		a-=(a&-a);	
	}	
	return ret;
}
 
void upd(ll a,ll x)
{
	while(a<=n)
	{
		pw[a]+=x;
		a+=(a&-a);
	}	
}
 
void updrange(ll a,ll sz)
{
	upd(a,qy);
	upd(a+sz,-qy);
	qy-=a;
}
 
ll dfs(ll a,ll p)
{
	ll ret=1;
	ind[a]=cnt++,par[ind[a]]=ind[p];
	for(int i=0;i<g[a].size();i++)
		ret+=dfs(g[a][i],a);
	chd[ind[a]]=ret;
	return ret;
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
	cin>>n>>q;
	for(int i=2;i<=n;i++)
	{
		ll a;cin>>a;
		g[a].push_back(i);
	}    
	dfs(1,0);
	for(int i=0;i<q;i++)
	{
		ll b,c,d;cin>>b>>c>>d;
		if(d==0)
		{
			if(val(ind[b])==val(ind[c]))
				cout<<"YES"<<"\n";
			else
				cout<<"NO"<<"\n";
		}
		else
		{
			if(val(ind[b])==val(ind[c]))
			{
				cout<<"YES"<<"\n";
				ll f=ind[b];
				if(val(f)==val(par[f]))
					updrange(f,chd[f]);
			}
			else
			{
				cout<<"NO"<<"\n";
				ll f=ind[c];
				if(val(f)==val(par[f]))
					updrange(f,chd[f]);	
			}
		}
	}
    return 0;
}

Compilation message

tree.cpp: In function 'll dfs(ll, ll)':
tree.cpp:42:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<g[a].size();i++)
              ~^~~~~~~~~~~~

Subtask #1
9.0 / 9.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5120 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	6 ms	5120 KB	Output is correct
4	Correct	6 ms	5248 KB	Output is correct
5	Correct	6 ms	5248 KB	Output is correct
6	Correct	7 ms	5248 KB	Output is correct
7	Correct	7 ms	5248 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	8 ms	5248 KB	Output is correct
10	Correct	6 ms	5248 KB	Output is correct

Subtask #2
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5120 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	6 ms	5120 KB	Output is correct
4	Correct	6 ms	5248 KB	Output is correct
5	Correct	6 ms	5248 KB	Output is correct
6	Correct	7 ms	5248 KB	Output is correct
7	Correct	7 ms	5248 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	8 ms	5248 KB	Output is correct
10	Correct	6 ms	5248 KB	Output is correct
11	Correct	9 ms	5120 KB	Output is correct
12	Correct	9 ms	5120 KB	Output is correct
13	Correct	7 ms	5120 KB	Output is correct
14	Correct	8 ms	5120 KB	Output is correct
15	Correct	6 ms	5120 KB	Output is correct
16	Correct	6 ms	5120 KB	Output is correct
17	Correct	6 ms	5120 KB	Output is correct
18	Correct	7 ms	5120 KB	Output is correct
19	Correct	9 ms	5120 KB	Output is correct
20	Correct	9 ms	5120 KB	Output is correct
21	Correct	7 ms	5120 KB	Output is correct

Subtask #3
14.0 / 14.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5120 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	6 ms	5120 KB	Output is correct
4	Correct	6 ms	5248 KB	Output is correct
5	Correct	6 ms	5248 KB	Output is correct
6	Correct	7 ms	5248 KB	Output is correct
7	Correct	7 ms	5248 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	8 ms	5248 KB	Output is correct
10	Correct	6 ms	5248 KB	Output is correct
11	Correct	9 ms	5120 KB	Output is correct
12	Correct	9 ms	5120 KB	Output is correct
13	Correct	7 ms	5120 KB	Output is correct
14	Correct	8 ms	5120 KB	Output is correct
15	Correct	6 ms	5120 KB	Output is correct
16	Correct	6 ms	5120 KB	Output is correct
17	Correct	6 ms	5120 KB	Output is correct
18	Correct	7 ms	5120 KB	Output is correct
19	Correct	9 ms	5120 KB	Output is correct
20	Correct	9 ms	5120 KB	Output is correct
21	Correct	7 ms	5120 KB	Output is correct
22	Correct	76 ms	5752 KB	Output is correct
23	Correct	76 ms	5864 KB	Output is correct
24	Correct	94 ms	5752 KB	Output is correct
25	Correct	127 ms	5872 KB	Output is correct
26	Correct	80 ms	5752 KB	Output is correct
27	Correct	82 ms	5880 KB	Output is correct
28	Correct	72 ms	5880 KB	Output is correct
29	Correct	79 ms	5952 KB	Output is correct
30	Correct	83 ms	6136 KB	Output is correct
31	Correct	122 ms	6008 KB	Output is correct

Subtask #4
0 / 19.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5120 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	6 ms	5120 KB	Output is correct
4	Correct	6 ms	5248 KB	Output is correct
5	Correct	6 ms	5248 KB	Output is correct
6	Correct	7 ms	5248 KB	Output is correct
7	Correct	7 ms	5248 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	8 ms	5248 KB	Output is correct
10	Correct	6 ms	5248 KB	Output is correct
11	Correct	273 ms	27512 KB	Output is correct
12	Correct	294 ms	27592 KB	Output is correct
13	Correct	273 ms	27512 KB	Output is correct
14	Correct	304 ms	27640 KB	Output is correct
15	Correct	315 ms	27512 KB	Output is correct
16	Correct	251 ms	27728 KB	Output is correct
17	Correct	233 ms	27768 KB	Output is correct
18	Incorrect	226 ms	27896 KB	Output isn't correct
19	Halted	0 ms	0 KB	-

Subtask #5
0 / 45.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5120 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	6 ms	5120 KB	Output is correct
4	Correct	6 ms	5248 KB	Output is correct
5	Correct	6 ms	5248 KB	Output is correct
6	Correct	7 ms	5248 KB	Output is correct
7	Correct	7 ms	5248 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	8 ms	5248 KB	Output is correct
10	Correct	6 ms	5248 KB	Output is correct
11	Correct	9 ms	5120 KB	Output is correct
12	Correct	9 ms	5120 KB	Output is correct
13	Correct	7 ms	5120 KB	Output is correct
14	Correct	8 ms	5120 KB	Output is correct
15	Correct	6 ms	5120 KB	Output is correct
16	Correct	6 ms	5120 KB	Output is correct
17	Correct	6 ms	5120 KB	Output is correct
18	Correct	7 ms	5120 KB	Output is correct
19	Correct	9 ms	5120 KB	Output is correct
20	Correct	9 ms	5120 KB	Output is correct
21	Correct	7 ms	5120 KB	Output is correct
22	Correct	76 ms	5752 KB	Output is correct
23	Correct	76 ms	5864 KB	Output is correct
24	Correct	94 ms	5752 KB	Output is correct
25	Correct	127 ms	5872 KB	Output is correct
26	Correct	80 ms	5752 KB	Output is correct
27	Correct	82 ms	5880 KB	Output is correct
28	Correct	72 ms	5880 KB	Output is correct
29	Correct	79 ms	5952 KB	Output is correct
30	Correct	83 ms	6136 KB	Output is correct
31	Correct	122 ms	6008 KB	Output is correct
32	Correct	273 ms	27512 KB	Output is correct
33	Correct	294 ms	27592 KB	Output is correct
34	Correct	273 ms	27512 KB	Output is correct
35	Correct	304 ms	27640 KB	Output is correct
36	Correct	315 ms	27512 KB	Output is correct
37	Correct	251 ms	27728 KB	Output is correct
38	Correct	233 ms	27768 KB	Output is correct
39	Incorrect	226 ms	27896 KB	Output isn't correct
40	Halted	0 ms	0 KB	-

Submission #107536

Compilation message

Subtask #1 9.0 / 9.0

Subtask #2 13.0 / 13.0

Subtask #3 14.0 / 14.0

Subtask #4 0 / 19.0

Subtask #5 0 / 45.0

Subtask #1
9.0 / 9.0

Subtask #2
13.0 / 13.0

Subtask #3
14.0 / 14.0

Subtask #4
0 / 19.0

Subtask #5
0 / 45.0