bzoj 3351 [ioi2009]Regions

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 #define mp  make_pair
  4 #define pii  pair<int,int>
  5 int const N = 200000 + 3;
  6 int const sz = 1000; 
  7 struct edge {
  8     int to, nt;
  9 } e[N << 1];
 10 int n, r, q, ans[N], v[N], cnt, h[N], num[N], ct[N], bl[N], c[10000][500], nb, id[N], nc;
 11 // 这里我们按照size分块
 12 // v 表示每个点的颜色
 13 // ans 表示答案
 14 // num 表示每个颜色出现的次数，也表示分块以后每个块里面点的数量
 15 // ct 表示每种颜色出现的次数
 16 // bl 表示每个点分别属于哪个块
 17 // c数组表示每个块里面超过sz的颜色的点的数量
 18 // nb 表示块的数量
 19 // id 表示超过sz的颜色的新编号
 20 // nc表示超过sz的颜色的数量
 21 // a vector表示小于等于sz的那些询问
 22 // b 表示小于等于sz的那些询问的编号
 23 // vex 记录哪个块里面有哪些点。
 24 // H 记录虚树
 25 // cp表示每一块的顶端节点
 26 int H[N], tin[N], tout[N], sum, cp[N], son[10000][500],yl[N],tq,pos[N];
 27 vector<int> a[N], b[N], vex[10000];
 28 map<pii, int> mat;
 29 void add(int a, int b) {
 30     e[++cnt].to = b;
 31     e[cnt].nt = h[a];
 32     h[a] = cnt;
 33 }
 34 int Add(int a, int b) {
 35     e[++cnt].to = b;
 36     e[cnt].nt = H[a];
 37     H[a] = cnt;
 38 }
 39 void dfs(int x) {
 40     num[v[x]]++;
 41     if(ct[v[x]] <= sz) {
 42         for(int i = 0; i < a[v[x]].size(); i++) {
 43             int t = a[v[x]][i];
 44             ans[b[v[x]][i]] += num[t];
 45         }
 46     }
 47     for(int i = h[x]; i; i = e[i].nt)
 48         dfs(e[i].to);
 49     num[v[x]]--;
 50 }
 51 void dfs2(int x, int fa) {
 52     tin[x] = ++sum;
 53     if(num[bl[fa]] >= sz) {
 54         nb++;
 55         cp[nb] = x;
 56         Add(bl[fa], nb);
 57     }
 58     bl[x] = nb;
 59     num[nb]++;
 60     if(ct[v[x]] > sz)
 61         son[nb][id[v[x]]]++;
 62     for(int i = h[x]; i; i = e[i].nt)
 63         dfs2(e[i].to, x);
 64     tout[x] = ++sum;
 65 }
 66 inline int ancestor(int x, int y) {
 67     return tin[x] <= tin[y] && tout[y] <= tout[x];
 68 }
 69 void dfs3(int x) {
 70     for(int i = H[x]; i; i = e[i].nt) {
 71         int s = e[i].to;
 72         dfs3(s);
 73         for(int j = 1; j <= nc; j++)
 74             son[x][j] += son[s][j];
 75     }
 76 }
 77 void ask(int x) {
 78     for(int i = 0; i < vex[x].size(); i++) {
 79         for(int j = H[x]; j; j = e[j].nt) {
 80             int a = vex[x][i];
 81             int b = cp[e[j].to];
 82             if(!ancestor(a, b))
 83                 continue;
 84             for(int cl = 1; cl <= nc; cl++) {
 85                 int t = mat[mp(v[a], pos[cl])];
 86                 if(!t)
 87                     continue;
 88                 ans[t] += son[e[j].to][cl];
 89             }
 90         }
 91     }
 92     for(int i = H[x]; i; i = e[i].nt)
 93         ask(e[i].to);
 94 }
 95 int main() {
 96     scanf("%d%d%d", &n, &r, &q);
 97     scanf("%d", &v[1]);
 98     ct[v[1]]++;
 99     for(int i = 2; i <= n; i++) {
100         int x, y;
101         scanf("%d%d", &x, &y);
102         add(x, i);
103         v[i] = y;
104         ct[v[i]]++;
105     }
106     for(int i = 1; i <= r; i++)
107         if(ct[i] > sz) {
108             id[i] = ++nc;
109             pos[nc]=i;
110         }
111     for(int i = 1; i <= q; i++) {
112         int x, y;
113         scanf("%d%d", &x, &y);
114         if(mat.find(mp(x,y))==mat.end()) {
115             mat[mp(x,y)]=++tq;
116             a[y].push_back(x);
117             b[y].push_back(tq);
118         }
119         yl[i]=mat[mp(x,y)];
120     }
121     cp[0] = 1;
122     dfs(1);
123     memset(num, 0, sizeof(num));
124     dfs2(1, 1);
125     for(int i = 1; i <= n; i++)
126         vex[bl[i]].push_back(i);
127     for(int i = 0; i <= nb; i++) {
128         for(int j = 0; j < vex[i].size(); j++)
129             for(int k = 0; k < vex[i].size(); k++) {
130                 int x = v[vex[i][j]];
131                 int y = v[vex[i][k]];
132                 if(vex[i][j]==vex[i][k])
133                     continue;
134                 if(!ancestor(vex[i][j], vex[i][k]))
135                     continue;
136                 if(ct[y] <= sz)
137                     continue;
138                 int t = mat[mp(x, y)];
139                 if(!t)
140                     continue;
141                 ans[t]++;
142             }
143     }
144     dfs3(0);
145     ask(0);
146     for(int i = 1; i <= q; i++)
147         printf("%d\n", ans[yl[i]]);
148     return 0;
149 }