Advertisement
Hippskill

Untitled

Jan 21st, 2016
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.55 KB | None | 0 0
  1. #define _CRT_SECURE_NO_WARNINGS
  2. #include<stdio.h>
  3. #include<iostream>
  4. #include<vector>
  5. #include<cmath>
  6. #include<algorithm>
  7. #include<memory.h>
  8. #include<map>
  9. #include<set>
  10. #include<queue>
  11. #include<list>
  12. #include<sstream>
  13. #include<cstring>
  14. #include<numeric>
  15. using namespace std;
  16.  
  17. const int N = 4e5 + 20;
  18.  
  19. vector<int> g[N];
  20.  
  21. int a[N];
  22. long long ans[4 * N];
  23. bool pushing[4 * N];
  24.  
  25. int in[N], out[N];
  26. int p[N];
  27. int idx = 0;
  28. int root = -1;
  29. bool u[N];
  30.  
  31.  
  32. int bitCount(long long v) {
  33.     int res = 0;
  34.     while (v > 0) {
  35.         if (v % 2 != 0) {
  36.             res++;
  37.         }
  38.         v /= 2;
  39.     }
  40.     return res;
  41. }
  42.  
  43. void push(int v) {
  44.     if (pushing[v]) {
  45.         ans[2 * v] = ans[2 * v + 1] = ans[v];
  46.         pushing[2 * v] = pushing[2 * v + 1] = true;
  47.         pushing[v] = false;
  48.     }
  49. }
  50.  
  51. void build(int v, int tl, int tr) {
  52.     if (tl >= tr) {
  53.         ans[v] = 1LL << a[p[tl]];
  54.         pushing[v] = true;
  55.     }
  56.     else {
  57.         int mid = (tl + tr) >> 1;
  58.         build(v * 2, tl, mid);
  59.         build(v * 2 + 1, mid + 1, tr);
  60.         ans[v] = ans[v * 2] | ans[v * 2 + 1];
  61.         pushing[v] = false;
  62.     }
  63. }
  64.  
  65. long long query(int v, int tl, int tr, int l, int r) {
  66.     if (tr < l || r < tl) {
  67.         return 0;
  68.     }
  69.     if (l <= tl && tr <= r) {
  70.         return ans[v];
  71.     }
  72.     push(v);
  73.     int mid = (tl + tr) >> 1;
  74.     long long res = query(v * 2, tl, mid, l, r);
  75.     res |= query(2 * v + 1, mid + 1, tr, l, r);
  76.     ans[v] = ans[2 * v] | ans[2 * v + 1];
  77.     return res;
  78. }
  79.  
  80. void upd(int v, int tl, int tr, int l, int r, long long c) {
  81.     if (tr < l || r < tl) {
  82.         return;
  83.     }
  84.     if (l <= tl && tr <= r) {
  85.         ans[v] = 1LL << c;
  86.         pushing[v] = true;
  87.         return;
  88.     }
  89.     push(v);
  90.     int mid = (tl + tr) >> 1;
  91.     upd(2 * v, tl, mid, l, r, c);
  92.     upd(2 * v + 1, mid + 1, tr, l, r, c);
  93.     ans[v] = ans[v * 2] | ans[v * 2 + 1];
  94. }
  95.  
  96. void dfs(int v) {
  97.     in[v] = idx;
  98.     p[idx++] = v;
  99.     u[v] = true;
  100.     for (auto to : g[v]) {
  101.         if (!u[to]) {
  102.             dfs(to);
  103.         }
  104.     }
  105.     out[v] = idx - 1;
  106. }
  107.  
  108. void addEdge(int x, int y) {
  109.     g[x].push_back(y);
  110.     g[y].push_back(x);
  111. }
  112.  
  113. int main() {
  114. #ifndef ONLINE_JUDGE
  115.     freopen("input.txt", "r", stdin);
  116. #endif
  117.     int n, m;
  118.     scanf("%d%d", &n, &m);
  119.     for (int i = 0; i < n; i++) {
  120.         scanf("%d", &a[i]);
  121.     }
  122.     for (int i = 0; i < n - 1; i++) {
  123.         int x, y;
  124.         scanf("%d%d", &x, &y);
  125.         addEdge(x - 1, y - 1);
  126.     }
  127.  
  128.  
  129.     root = 0;
  130.     dfs(root);
  131.     build(1, 0, n - 1);
  132.  
  133.     for (int i = 0; i < m; i++) {
  134.         int t;
  135.         scanf("%d", &t);
  136.         if (t == 1) {
  137.             int v, c;
  138.             scanf("%d%d", &v, &c);
  139.             upd(1, 0, n - 1, in[v - 1], out[v - 1], c);
  140.         }
  141.         else {
  142.             int v;
  143.             scanf("%d", &v);
  144.             printf("%d\n", bitCount(query(1, 0, n - 1, in[v - 1], out[v - 1])));
  145.         }
  146.     }
  147.  
  148.     return 0;
  149. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement