HDU5039--Hilarity DFS序+线段树区间更新 14年北京网络赛
题意：n个点的树，每个条边权值为0或者1， q次操作
Q 路径边权抑或和为1的点对数， （u， v）（v, u）算2个。
M i修改第i条边的权值 如果是0则变成1， 否则变成0
作法： 我们可以求出每个点到根节点路径边权抑或和为val， 那么ans = val等于0的个数乘val等于1的个数再乘2。
注意到每一次修改操作，只会影响以u为根的子树（假设边为u----v dep[v] > dep[u]）， 那么每次只需把子树区间的值与1抑或就行了。 这一步可以用线段树区间更新。
比赛时过的人好少。。。好奇怪。
1 #include <bits/stdc++.h> 2 using namespace std; 3 const int MAXN = 3e4 + 1; 4 struct Edge { 5 int to, cost; 6 Edge (int to, int cost){ 7 this->to = to, this->cost = cost; 8 } 9 }; 10 vector <Edge> G[MAXN]; 11 int val[MAXN]; 12 int siz[MAXN], pt1[MAXN], pt2[MAXN], IDX, dep[MAXN]; 13 void dfs(int u, int father, int k) { 14 val[u] = k; 15 dep[u] = dep[father] + 1; 16 pt1[u] = ++IDX; 17 for (Edge e: G[u]) { 18 int v = e.to; 19 if (v != father) { 20 dfs(v, u, k^e.cost); 21 } 22 } 23 pt2[u] = IDX; 24 } 25 int seg[2][MAXN << 2], lazy[MAXN << 2]; 26 void push_down (int pos) { 27 if (lazy[pos]) { 28 swap(seg[0][pos<<1], seg[1][pos<<1]); 29 swap(seg[0][pos<<1|1], seg[1][pos<<1|1]); 30 lazy[pos<<1] ^= lazy[pos]; 31 lazy[pos<<1|1] ^= lazy[pos]; 32 lazy[pos] = 0; 33 } 34 } 35 void build (int l, int r, int pos, int x, int d) { 36 if (l == r) { 37 seg[0][pos] = d == 0; 38 seg[1][pos] = d == 1; 39 return; 40 } 41 int mid = (l + r) >> 1; 42 if (x <= mid) { 43 build(l, mid, pos<<1, x, d); 44 } else { 45 build(mid+1, r, pos<<1|1, x, d); 46 } 47 seg[0][pos] = seg[0][pos<<1] + seg[0][pos<<1|1]; 48 seg[1][pos] = seg[1][pos<<1] + seg[1][pos<<1|1]; 49 } 50 void update (int l, int r, int pos, int ua, int ub) { 51 if (ua <= l && ub >= r) { 52 lazy[pos] ^= 1; 53 swap(seg[0][pos], seg[1][pos]); 54 return; 55 } 56 push_down(pos); 57 int mid = (l + r) >> 1; 58 if (ua <= mid) { 59 update(l, mid, pos<<1, ua, ub); 60 } 61 if (ub > mid) { 62 update(mid+1, r, pos<<1|1, ua, ub); 63 } 64 seg[0][pos] = seg[0][pos<<1] + seg[0][pos<<1|1]; 65 seg[1][pos] = seg[1][pos<<1] + seg[1][pos<<1|1]; 66 } 67 void init () { 68 IDX = 0; 69 memset(lazy, 0, sizeof (lazy)); 70 memset(seg, 0, sizeof seg); 71 for (int i = 0; i < MAXN; i++) { 72 G[i].clear(); 73 } 74 } 75 pair <int, int> edge[MAXN]; 76 int main() { 77 #ifndef ONLINE_JUDGE 78 freopen("in.txt", "r", stdin); 79 #endif // ONLINE_JUDGE 80 int T, cas = 1; 81 scanf ("%d", &T); 82 while (T--) { 83 init(); 84 int n, q; 85 int tot = 0; 86 map <string, int> mp; 87 scanf ("%d", &n); 88 for (int i = 1; i <= n; i++) { 89 char buff[15]; 90 scanf ("%s", buff); 91 mp[buff] = ++tot; 92 } 93 for (int i = 0; i < n-1; i++) { 94 char name1[15], name2[15]; 95 int status; 96 scanf ("%s%s%d", name1, name2, &status); 97 edge[i] = make_pair(mp[name1], mp[name2]); 98 G[edge[i].first].push_back(Edge(edge[i].second, status)); 99 G[edge[i].second].push_back(Edge(edge[i].first, status)); 100 } 101 dfs(1, 0, 0); 102 build(1, IDX, 1, pt1[1], 0); 103 for (int i = 0; i < n-1; i++) { 104 int u = edge[i].first; 105 int v = edge[i].second; 106 if (dep[u] > dep[v]) { 107 swap(u, v); 108 } 109 build(1, IDX, 1, pt1[v], val[v]); 110 } 111 scanf ("%d", &q); 112 printf("Case #%d:\n", cas++); 113 for (int i = 0; i < q; i++) { 114 char kind[5]; 115 scanf ("%s", kind); 116 if (kind[0] == 'Q') { 117 printf("%d\n", seg[0][1] * seg[1][1] * 2); 118 } else { 119 int e; 120 scanf ("%d", &e); 121 int u = edge[e-1].first; 122 int v = edge[e-1].second; 123 if (dep[u] > dep[v]) { 124 swap(u, v); 125 } 126 update(1, IDX, 1, pt1[v], pt2[v]); 127 } 128 } 129 } 130 return 0; 131 }
