dfs序

 

dfs序可以把一棵树转化为线性结构,便于区间操作

dfs序的代码：

#include<bits/stdc++.h>
#define MAXN 100005
using namespace std;
vector<int> vc[MAXN];
int in[MAXN],out[MAXN];
int time1=0;
int arr[MAXN];
//把树转化成线性序列
//对区间(in[x],out[x]),操作即对以x为根结点的子树操作

void dfs(int x,int fa)
{
    in[x]=++time1;
    arr[time1]=x;
    for(int i=0;i<vc[x].size();i++)
    {
        int v=vc[x][i];
        if(v==fa) continue;
        dfs(v,x);
    }
    out[x]=time1;
}


int main()
{
    int n,m,a,b;
    scanf("%d%d",&n,&m);
    while(m--)
    {
        scanf("%d%d",&a,&b);
        vc[a].push_back(b);
    }
    dfs(1,-1);
    for(int i=1;i<=n;i++)
        printf("%d ",arr[i]);
    printf("\n");
    for(int i=1;i<=n;i++)
        printf("%d:%d %d\n",i,in[i],out[i]);

    return 0;
}

对区间(in[x],out[x]),操作即对以x为根结点的子树操作

POJ - 3321

题意:

给定一棵树,有两种操作

Q x:询问子树x的结点和

C x:如果x结点为1,则变0,如果为0则变1

解析:dfs序+树状数组

查询用树状数组求out[i]~in[i]-1的和

修改用树状数组单点修改in[i]的值

ac:

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lowbit(x) x&(-x)
#define MAXN 500005
using namespace std;
int head[MAXN];
int to[MAXN],next1[MAXN];
int tot=0,time=0;
int in[MAXN],out[MAXN];
int gg[MAXN],have[MAXN];

void add(int u,int v)
{
    to[tot]=v;
    next1[tot]=head[u];
    head[u]=tot++;
}

void dfs(int x)
{
    in[x]=++time;
    for(int i=head[x];i!=-1;i=next1[i])
        dfs(to[i]);
    out[x]=time;
}

void update(int x,int v)
{
    while(x<=time)
    {
        gg[x]+=v;
        x+=lowbit(x);
    }
}

int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans+=gg[x];
        x-=lowbit(x);
    }
    return ans;
}

char cc[5];

int main()
{
    memset(head,-1,sizeof(head));
    int n,a,b,m,g,x;
    scanf("%d",&n);
    for(int i=0;i<n-1;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    dfs(1);
    for(int i=1;i<=n;i++)
    {
        have[i]=1;
        update(in[i],1);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s %d",&cc,&x);
        if(cc[0]=='C')
        {
            if(have[x]==1)
                update(in[x],-1);
            else
                update(in[x],1);
            have[x]=1-have[x];
        }
        else{
            printf("%d\n",sum(out[x])-sum(in[x]-1));
        }
    }
    return 0;
}