Poj3221 dfs序

来源：互联网发布：java获取客户端端口号编辑：程序博客网时间：2024/06/07 10:18

利用dfs序的子树连续的性质,用线段树单点修改,区间查询
PS:与子树修改,路径查询的dfs不同,一般不需要记录出栈序,只需要记录出栈位置

#include<iostream>#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fod(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=2e5+10;int ls[N<<2],rs[N<<2],sum[N<<2],n,m,seq[N],st[N],ed[N],cnt=0,head[N],tot=0;struct Edge{    int to,next;    Edge(int to=0,int next=0):to(to),next(next){}}e[N<<1];void add_edge(int u,int v){e[++tot]=Edge(v,head[u]);head[u]=tot;}void dfs(int u,int fa){    seq[st[u]=++cnt]=u;    for(int i=head[u];i;i=e[i].next){        int id=e[i].to;        if(id==fa)continue;        dfs(id,u);     }    ed[u]=cnt;}void build(int k,int l,int r){    ls[k]=l;rs[k]=r;sum[k]=0;    if(l==r){sum[k]=1;return;}    int mid=(l+r)>>1;    build(k<<1,l,mid);    build(k<<1|1,mid+1,r);    sum[k]=sum[k<<1]+sum[k<<1|1];}void update(int k,int pos){    if(ls[k]==rs[k]){sum[k]^=1;return;}    int mid=(ls[k]+rs[k])>>1;    if(pos<=mid)update(k<<1,pos);    else update(k<<1|1,pos);    sum[k]=sum[k<<1]+sum[k<<1|1];}int query(int k,int ql,int qr){    if(ls[k]==ql&&rs[k]==qr){return sum[k];}    int mid=(ls[k]+rs[k])>>1;    if(qr<=mid)return query(k<<1,ql,qr);    else if(ql>mid)return query(k<<1|1,ql,qr);    else return (query(k<<1,ql,mid)+query(k<<1|1,mid+1,qr));}int main(){    scanf("%d",&n);    for(int u,v,i=1;i<n;i++) {        scanf("%d%d",&u,&v);        add_edge(u,v);add_edge(v,u);    }dfs(1,0);    build(1,1,n);    scanf("%d",&m);    for(int i=1;i<=m;i++) {        char op[3];int x;        scanf("%s%d",op,&x);        if(op[0]=='C'){update(1,st[x]);}        else {printf("%d\n",query(1,st[x],ed[x]));}    }    return 0;}

阅读全文

0 0