Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <bits/stdc++.h>
- #pragma comment(linker, "/STACK:1000000000")
- #define int long long
- #define mp make_pair
- #define pb push_back
- using namespace std;
- vector<vector<int> > g;
- vector<int> color;
- vector<int> res;
- //vector<int> used;
- set<int> dfs(int start) {
- int to;
- //used[start] = 1;
- set<int> s1, s2;
- s1.insert(color[start]);
- for (int i = 0; i < g[start].size(); i++) {
- to = g[start][i];
- //if (!used[to]) {
- s2 = dfs(to);
- if (s1.size() < s2.size())
- s1.swap(s2);
- s1.insert(s2.begin(), s2.end());
- s2.clear();
- //}
- }
- res[start] = s1.size();
- return s1;
- }
- signed main() {
- ios_base::sync_with_stdio(0);
- cin.tie(0);
- int n;
- cin >> n;
- g = vector<vector<int> >(n);
- color = vector<int>(n);
- res = vector<int>(n);
- //used = vector<int>(n);
- int u, v;
- int root = -1;
- for (int i = 0; i < n; ++i) {
- cin >> u >> v;
- u--;
- color[i] = v;
- if (u == -1) {
- root = i;
- continue;
- }
- g[u].pb(i);
- }
- dfs(root);
- for (auto i : res) {
- cout << i << ' ';
- }
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement