Submission #107544

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
107544	2019-04-25T01:55:52 Z	aer0park	None (KOI16_tree)	C++14	36 / 100	290 ms	27768 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	5248 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	7 ms	5248 KB	Output is correct
4	Correct	7 ms	5120 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	5120 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	6 ms	5120 KB	Output is correct
10	Correct	7 ms	5120 KB	Output is correct

Subtask #2
13.0 / 13.0

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

Subtask #3
14.0 / 14.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5248 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	7 ms	5248 KB	Output is correct
4	Correct	7 ms	5120 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	5120 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	6 ms	5120 KB	Output is correct
10	Correct	7 ms	5120 KB	Output is correct
11	Correct	8 ms	5120 KB	Output is correct
12	Correct	6 ms	5120 KB	Output is correct
13	Correct	6 ms	5120 KB	Output is correct
14	Correct	7 ms	5120 KB	Output is correct
15	Correct	7 ms	5120 KB	Output is correct
16	Correct	8 ms	5120 KB	Output is correct
17	Correct	7 ms	5120 KB	Output is correct
18	Correct	10 ms	5120 KB	Output is correct
19	Correct	8 ms	5120 KB	Output is correct
20	Correct	7 ms	5120 KB	Output is correct
21	Correct	6 ms	5120 KB	Output is correct
22	Correct	80 ms	5792 KB	Output is correct
23	Correct	83 ms	5844 KB	Output is correct
24	Correct	78 ms	5872 KB	Output is correct
25	Correct	70 ms	5752 KB	Output is correct
26	Correct	80 ms	5752 KB	Output is correct
27	Correct	77 ms	5840 KB	Output is correct
28	Correct	68 ms	5880 KB	Output is correct
29	Correct	70 ms	5880 KB	Output is correct
30	Correct	74 ms	6040 KB	Output is correct
31	Correct	67 ms	6004 KB	Output is correct

Subtask #4
0 / 19.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	5248 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	7 ms	5248 KB	Output is correct
4	Correct	7 ms	5120 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	5120 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	6 ms	5120 KB	Output is correct
10	Correct	7 ms	5120 KB	Output is correct
11	Correct	260 ms	27564 KB	Output is correct
12	Correct	243 ms	27512 KB	Output is correct
13	Correct	290 ms	27768 KB	Output is correct
14	Correct	242 ms	27640 KB	Output is correct
15	Correct	277 ms	27656 KB	Output is correct
16	Correct	270 ms	27688 KB	Output is correct
17	Correct	261 ms	27768 KB	Output is correct
18	Incorrect	260 ms	27768 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	5248 KB	Output is correct
2	Correct	6 ms	5120 KB	Output is correct
3	Correct	7 ms	5248 KB	Output is correct
4	Correct	7 ms	5120 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	5120 KB	Output is correct
8	Correct	6 ms	5248 KB	Output is correct
9	Correct	6 ms	5120 KB	Output is correct
10	Correct	7 ms	5120 KB	Output is correct
11	Correct	8 ms	5120 KB	Output is correct
12	Correct	6 ms	5120 KB	Output is correct
13	Correct	6 ms	5120 KB	Output is correct
14	Correct	7 ms	5120 KB	Output is correct
15	Correct	7 ms	5120 KB	Output is correct
16	Correct	8 ms	5120 KB	Output is correct
17	Correct	7 ms	5120 KB	Output is correct
18	Correct	10 ms	5120 KB	Output is correct
19	Correct	8 ms	5120 KB	Output is correct
20	Correct	7 ms	5120 KB	Output is correct
21	Correct	6 ms	5120 KB	Output is correct
22	Correct	80 ms	5792 KB	Output is correct
23	Correct	83 ms	5844 KB	Output is correct
24	Correct	78 ms	5872 KB	Output is correct
25	Correct	70 ms	5752 KB	Output is correct
26	Correct	80 ms	5752 KB	Output is correct
27	Correct	77 ms	5840 KB	Output is correct
28	Correct	68 ms	5880 KB	Output is correct
29	Correct	70 ms	5880 KB	Output is correct
30	Correct	74 ms	6040 KB	Output is correct
31	Correct	67 ms	6004 KB	Output is correct
32	Correct	260 ms	27564 KB	Output is correct
33	Correct	243 ms	27512 KB	Output is correct
34	Correct	290 ms	27768 KB	Output is correct
35	Correct	242 ms	27640 KB	Output is correct
36	Correct	277 ms	27656 KB	Output is correct
37	Correct	270 ms	27688 KB	Output is correct
38	Correct	261 ms	27768 KB	Output is correct
39	Incorrect	260 ms	27768 KB	Output isn't correct
40	Halted	0 ms	0 KB	-

Submission #107544

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