bzoj 1103（DFS序+树状数组）

最新推荐文章于 2019-08-07 21:06:38 发布

原创最新推荐文章于 2019-08-07 21:06:38 发布 · 381 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#DFS序 #树状数组

OI-数据结构专栏收录该内容

76 篇文章

订阅专栏

本文介绍了一种结合DFS序和树状数组的数据结构应用技巧，用于高效处理图上的动态更新与查询问题。具体实现包括如何通过DFS遍历获得节点的进入与离开时间，并利用树状数组快速更新和查询特定节点路径上的边权之和。

传送门
题意：
两种操作：
1.修改某条边的边权（0/1）
2.询问某个点到1路径上的边权和

题解：
在DFS序上差分+树状数组。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=250004;
int n,m;
int head[MAXN],etot=0;
struct EDGE {
    int v,nxt;
}e[MAXN<<1];
int in[MAXN],out[MAXN],tim=0;

inline int read() {
    int x=0;char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x;
}
inline void adde(int u,int v) {
    e[etot].nxt=head[u],e[etot].v=v,head[u]=etot++;
    e[etot].nxt=head[v],e[etot].v=u,head[v]=etot++;
}
void dfs(int p,int fa) {
    in[p]=++tim;
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa) dfs(v,p);
    }
    out[p]=tim;
}
int c[MAXN];
inline void add(int x,int delta) {
    for (;x<=n;x+=x&-x) c[x]+=delta;
}
inline int query(int x) {
    int ret=0;
    for (;x;x-=x&-x) ret+=c[x];
    return ret;
}
int main() {
//  freopen("bzoj 1103.in","r",stdin);
    memset(head,-1,sizeof(head));
    n=read();
    for (register int i=1;i<n;++i) {
        int u=read(),v=read();
        adde(u,v);
    }
    m=read()+n-1;
    dfs(1,0);
    for (register int i=2;i<=n;++i) add(in[i],1),add(out[i]+1,-1);
    for (register int i=0;i<m;++i) {
        char s;
        while (!isalpha(s=getchar()));
        if (s=='A') {
            int x=read(),y=read();
            add(in[y],-1),add(out[y]+1,1);
        } else {
            int x=read();
            printf("%d\n",query(in[x]));
        }
    }
    return 0;
}