Submission #30885

#	Time	Username	Problem	Language	Result	Execution time	Memory
30885		cgiosy	트리 (KOI16_tree)	C++14	0 / 100	26 ms	12960 KiB

tree

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

int n, m, q, a, b, c, d, x, k;
struct {int p; int r; set<int> c;} t[200010];

void f(int i)
{
	t[i].r = k;

	for(int j : t[i].c)
		f(j);
}

int main()
{
	cin >> n >> q;

	t[1].p = t[1].r = 1;

	for(int i = 2; i <= n; i++)
	{
		cin >> a;
		t[i].p = a;
		t[i].r = 1;
		t[a].c.insert(i);
	}

	for(int i = 0, m = n + q; i < q; i++)
	{
		cin >> b >> c >> d;
		x = (t[b].r == t[c].r);

		puts(x ? "YES" : "NO");

		if(d)
		{
			k = x ? b : c;
			t[k].c.erase(t[k].p);
			f(k);
		}
	}
}

Compilation message (stderr)

tree.cpp: In function 'int main()':
tree.cpp:29:17: warning: unused variable 'm' [-Wunused-variable]
  for(int i = 0, m = n + q; i < q; i++)
                 ^

• Subtask 1: 0 / 9
• Subtask 2: 0 / 13
• Subtask 3: 0 / 14
• Subtask 4: 0 / 19
• Subtask 5: 0 / 45