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抑或就行了。 这一步可以用线段树区间更新。

比赛时过的人好少。。。好奇怪。

 #include <bits/stdc++.h>
 using namespace std;
 const int MAXN = 3e4 + ;
 struct Edge {
     int to, cost;
     Edge (int to, int cost){
         this->to = to, this->cost = cost;
     }
 };
 vector <Edge> G[MAXN];
 int val[MAXN];
 int siz[MAXN], pt1[MAXN], pt2[MAXN], IDX, dep[MAXN];
 void dfs(int u, int father, int k) {
     val[u] = k;
     dep[u] = dep[father] + ;
     pt1[u] = ++IDX;
     for (Edge e: G[u]) {
         int v = e.to;
         if (v != father) {
             dfs(v, u, k^e.cost);
         }
     }
     pt2[u] = IDX;
 }
 int seg[][MAXN << ], lazy[MAXN << ];
 void push_down (int pos) {
     if (lazy[pos]) {
         swap(seg[][pos<<], seg[][pos<<]);
         swap(seg[][pos<<|], seg[][pos<<|]);
         lazy[pos<<] ^= lazy[pos];
         lazy[pos<<|] ^= lazy[pos];
         lazy[pos] = ;
     }
 }
 void build (int l, int r, int pos, int x, int d) {
     if (l == r) {
         seg[][pos] = d == ;
         seg[][pos] = d == ;
         return;
     }
     int mid = (l + r) >> ;
     if (x <= mid) {
         build(l, mid, pos<<, x, d);
     } else {
         build(mid+, r, pos<<|, x, d);
     }
     seg[][pos] = seg[][pos<<] + seg[][pos<<|];
     seg[][pos] = seg[][pos<<] + seg[][pos<<|];
 }
 void update (int l, int r, int pos, int ua, int ub) {
     if (ua <= l && ub >= r) {
         lazy[pos] ^= ;
         swap(seg[][pos], seg[][pos]);
         return;
     }
     push_down(pos);
     int mid = (l + r) >> ;
     if (ua <= mid) {
         update(l, mid, pos<<, ua, ub);
     }
     if (ub > mid) {
         update(mid+, r, pos<<|, ua, ub);
     }
     seg[][pos] = seg[][pos<<] + seg[][pos<<|];
     seg[][pos] = seg[][pos<<] + seg[][pos<<|];
 }
 void init () {
     IDX = ;
     memset(lazy, , sizeof (lazy));
     memset(seg, , sizeof seg);
     for (int i = ; i < MAXN; i++) {
         G[i].clear();
     }
 }
 pair <int, int> edge[MAXN];
 int main() {
 #ifndef ONLINE_JUDGE
     freopen("in.txt", "r", stdin);
 #endif // ONLINE_JUDGE
     int T, cas = ;
     scanf ("%d", &T);
     while (T--) {
         init();
         int n, q;
         int tot = ;
         map <string, int> mp;
         scanf ("%d", &n);
         for (int i = ; i <= n; i++) {
             char buff[];
             scanf ("%s", buff);
             mp[buff] = ++tot;
         }
         for (int i = ; i < n-; i++) {
             char name1[], name2[];
             int status;
             scanf ("%s%s%d", name1, name2, &status);
             edge[i] = make_pair(mp[name1], mp[name2]);
             G[edge[i].first].push_back(Edge(edge[i].second, status));
             G[edge[i].second].push_back(Edge(edge[i].first, status));
         }
         dfs(, , );
         build(, IDX, , pt1[], );
         for (int i = ; i < n-; i++) {
             int u = edge[i].first;
             int v = edge[i].second;
             if (dep[u] > dep[v]) {
                 swap(u, v);
             }
             build(, IDX, , pt1[v], val[v]);
         }
         scanf ("%d", &q);
         printf("Case #%d:\n", cas++);
         for (int i = ; i < q; i++) {
             char kind[];
             scanf ("%s", kind);
             if (kind[] == 'Q') {
                 printf("%d\n", seg[][] * seg[][] * );
             } else {
                 int e;
                 scanf ("%d", &e);
                 int u = edge[e-].first;
                 int v = edge[e-].second;
                 if (dep[u] > dep[v]) {
                     swap(u, v);
                 }
                 update(, IDX, , pt1[v], pt2[v]);
             }
         }
     }
     return ;
 }