Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- //Nguyen Huu Hoang Minh
- #include <bits/stdc++.h>
- #define sz(x) int(x.size())
- #define all(x) x.begin(),x.end()
- #define reset(x) memset(x, 0,sizeof(x))
- #define pb push_back
- #define mp make_pair
- #define fi first
- #define se second
- #define N 100005
- #define remain(x) if (x > MOD) x -= MOD
- #define ii pair<int, int>
- #define iiii pair< ii , ii >
- #define viiii vector< iiii >
- #define vi vector<int>
- #define vii vector< ii >
- #define bit(x, i) (((x) >> (i)) & 1)
- #define Task "test"
- #define int long long
- using namespace std;
- typedef long double ld;
- const int inf = 1e10;
- const int minf = -1e10;
- int n, root, q;
- vector<int> g[N];
- int c[N];
- set<int> s[N];
- int res[N];
- void readfile()
- {
- ios_base::sync_with_stdio(false);
- cin.tie(0);cout.tie(0);
- if (fopen(Task".inp","r"))
- {
- freopen(Task".inp","r",stdin);
- //freopen(Task".out","w",stdout);
- }
- cin >> n >> q >> root;
- for(int i=1; i<n; i++){
- int u, v;
- cin >> u >> v;
- g[u].pb(v);
- g[v].pb(u);
- }
- for(int i=1; i<=n; i++) cin >> c[i];
- }
- void dfs(int u, int par){
- s[u].insert(c[u]);
- for(auto v : g[u]){
- if (v!=par){
- dfs(v,u);
- if (s[v].size()>s[u].size()) swap(s[u],s[v]);
- for(int i : s[v]) s[u].insert(i);
- }
- }
- res[u] = s[u].size();
- }
- void proc()
- {
- dfs(root,root);
- while (q--){
- int S; cin >> S;
- cout << res[S] << '\n';
- }
- }
- signed main()
- {
- readfile();
- proc();
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement