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 + 1;
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] + 1;
    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[2][MAXN << 2], lazy[MAXN << 2];
void push_down (int pos) {
    if (lazy[pos]) {
        swap(seg[0][pos<<1], seg[1][pos<<1]);
        swap(seg[0][pos<<1|1], seg[1][pos<<1|1]);
        lazy[pos<<1] ^= lazy[pos];
        lazy[pos<<1|1] ^= lazy[pos];
        lazy[pos] = 0;
    }
}
void build (int l, int r, int pos, int x, int d) {
    if (l == r) {
        seg[0][pos] = d == 0;
        seg[1][pos] = d == 1;
        return;
    }
    int mid = (l + r) >> 1;
    if (x <= mid) {
        build(l, mid, pos<<1, x, d);
    } else {
        build(mid+1, r, pos<<1|1, x, d);
    }
    seg[0][pos] = seg[0][pos<<1] + seg[0][pos<<1|1];
    seg[1][pos] = seg[1][pos<<1] + seg[1][pos<<1|1];
}
void update (int l, int r, int pos, int ua, int ub) {
    if (ua <= l && ub >= r) {
        lazy[pos] ^= 1;
        swap(seg[0][pos], seg[1][pos]);
        return;
    }
    push_down(pos);
    int mid = (l + r) >> 1;
    if (ua <= mid) {
        update(l, mid, pos<<1, ua, ub);
    }
    if (ub > mid) {
        update(mid+1, r, pos<<1|1, ua, ub);
    }
    seg[0][pos] = seg[0][pos<<1] + seg[0][pos<<1|1];
    seg[1][pos] = seg[1][pos<<1] + seg[1][pos<<1|1];
}
void init () {
    IDX = 0;
    memset(lazy, 0, sizeof (lazy));
    memset(seg, 0, sizeof seg);
    for (int i = 0; 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 = 1;
    scanf ("%d", &T);
    while (T--) {
        init();
        int n, q;
        int tot = 0;
        map <string, int> mp;
        scanf ("%d", &n);
        for (int i = 1; i <= n; i++) {
            char buff[15];
            scanf ("%s", buff);
            mp[buff] = ++tot;
        }
        for (int i = 0; i < n-1; i++) {
            char name1[15], name2[15];
            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(1, 0, 0);
        build(1, IDX, 1, pt1[1], 0);
        for (int i = 0; i < n-1; i++) {
            int u = edge[i].first;
            int v = edge[i].second;
            if (dep[u] > dep[v]) {
                swap(u, v);
            }
            build(1, IDX, 1, pt1[v], val[v]);
        }
        scanf ("%d", &q);
        printf("Case #%d:
", cas++);
        for (int i = 0; i < q; i++) {
            char kind[5];
            scanf ("%s", kind);
            if (kind[0] == 'Q') {
                printf("%d
", seg[0][1] * seg[1][1] * 2);
            } else {
                int e;
                scanf ("%d", &e);
                int u = edge[e-1].first;
                int v = edge[e-1].second;
                if (dep[u] > dep[v]) {
                    swap(u, v);
                }
                update(1, IDX, 1, pt1[v], pt2[v]);
            }
        }
    }
    return 0;
}