复习:自己以前瞎写的树dfs序

Posted saigyouji-yuyuko

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了复习:自己以前瞎写的树dfs序相关的知识,希望对你有一定的参考价值。

半年前的内容,2月12日前来考古。

这里枚举了树的DFS序来解决树上问题的多个板子,自己最好多看看。

↓改↓↓求↓
————————>>>这个就算了
简单, BIT
重点!
简单, 线段树
重重点!!!
简单, BIT+差分
重重点!!!
重重点!!!
重重点!!!

给定一颗树, 和每个节点的权值

1.对某个节点X权值加上一个数W, 查询某个子树X里所有点权的和

改点求树

这个不说什么,果断就BIT维护区间和,即改点求段。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100000;
inline int Read(){
    int x=0,f=0;char ch;
    while(!isdigit(ch=getchar()))if(ch==‘-‘) f=1;
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^‘0‘),ch=getchar();
    return f?-x:x;
}
vector<int> tree[MAXN];
int seq[MAXN],st[MAXN],ed[MAXN];
int n,x,y,cnt,q,m;

inline int lowbit(int x){return x&-x;}
inline void Add(int x,int k){while(x<=n) seq[x]+=k,x+=lowbit(x);}
inline int Sum(int x){int ans=0;while(x>0) ans+=seq[x],x-=lowbit(x);return ans;}

void DFS(int fa,int rt){
    st[rt]=++cnt;
    for(register int i=0;i<tree[rt].size();++i) if(tree[rt][i]!=fa) DFS(rt,tree[rt][i]);
    ed[rt]=cnt;
}

int main(){
    n=Read(),m=Read();//root------1
    for(register int i=1;i<n;++i) x=Read(),y=Read(),tree[x].push_back(y),tree[y].push_back(x);
    DFS(0,1);
    for(register int i=1;i<=n;++i) y=Read(),Add(st[i],y);
    for(register int i=1;i<=m;++i){
        q=Read();x=Read();
        if(q==1) y=Read(),Add(st[x],y);
        else printf("%d
",Sum(ed[x])-Sum(st[x]-1));
    }
    return 0;
}

2.对节点X到Y的最短路上所有点权都加一个数W, 查询某个点的权值

改链求点

思路:操作等价于是:

1.对X到根节点路径上所有点权加W
2.对Y到根节点路径上所有点权加W
3.对LCA(x, y)到根节点路径上所有点权值减W
4.对LCA(x,y)的父节点 parent(LCA(x,y))parent(LCA(x, y))parent(LCA(x,y))到根节点路径上所有权值减W

这里的操作我 用一个树状数组seq[i]seq[i]seq[i]表示在DFS序里的节点i到根节点的整体被操作的值,这样操作1就是Add(st[x],w)Add(st[x],w)Add(st[x],w) 2,3,4同理。查询自己看代码想一想,就是子树中总共的操作值。(感谢@hankeke帮助juruo理解)

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100000;
const int Log=19;
inline int Read(){
    int x=0,f=0;char ch;
    while(!isdigit(ch=getchar()))if(ch==‘-‘) f=1;
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^‘0‘),ch=getchar();
    return f?-x:x;
}
vector<int> tree[MAXN];
int f[MAXN][Log+2];
int seq[MAXN],st[MAXN],ed[MAXN],A[MAXN],depth[MAXN],s[MAXN];
int n,x,y,cnt,q,m,w;

inline int lowbit(int x){return x&-x;}
inline void Add(int x,int k){if(!x) return; while(x<=n) seq[x]+=k,x+=lowbit(x);}
inline int Sum(int x){int ans=0;while(x>0) ans+=seq[x],x-=lowbit(x);return ans;}

void Power(int k,int tot){
    while(k<=n+1){
        int kk=k;k<<=1;
        for(register int i=kk;i<=k-1;++i) s[i]=tot;
        ++tot;
    }
}

void DFS(int fa,int rt){
    st[rt]=++cnt;f[rt][0]=fa;depth[rt]=depth[fa]+1;
    for(register int j=1;(1<<j)<=depth[rt];++j) f[rt][j]=f[f[rt][j-1]][j-1];
    for(register int i=0;i<tree[rt].size();++i) if(tree[rt][i]!=fa) DFS(rt,tree[rt][i]);
    ed[rt]=cnt;
}

inline int LCA(int x,int y){
    if(depth[x]<depth[y]) x^=y^=x^=y;
    for(register int k=/*s[depth[x]-depth[y]]*/19;k>=0;--k) if(depth[f[x][k]]>=depth[y]) x=f[x][k];
    if(x==y) return x;
    for(register int k=/*s[depth[x]-1]*/19;k>=0;--k) if(f[x][k]!=f[y][k]) x=f[x][k],y=f[y][k];
    return f[x][0];
}

int main(){//freopen("tmp.txt","r",stdin);
    n=Read(),m=Read();//root------1
    for(register int i=1;i<n;++i) x=Read(),y=Read(),tree[x].push_back(y),tree[y].push_back(x);
    DFS(0,1);//Power(1,0);
    for(register int i=1;i<=n;++i) A[i]=Read();
    for(register int i=1;i<=m;++i){
        q=Read();
        if(q==1){
            x=Read(),y=Read(),w=Read();
            int lca=LCA(x,y);
            Add(st[x],w);Add(st[y],w);Add(st[lca],-w);Add(st[f[lca][0]],-w);
        }
        else x=Read(),printf("%d
",A[x]+Sum(ed[x])-Sum(st[x]-1));
    }
    return 0;
}

3.对节点X到Y的最短路上所有点权都加一个数W, 查询某个点子树的权值之和

改链求树

思路极乱

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100000;
const int Log=19;
inline int Read(){
    int x=0,f=0;char ch;
    while(!isdigit(ch=getchar()))if(ch==‘-‘) f=1;
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^‘0‘),ch=getchar();
    return f?-x:x;
}
vector<int> tree[MAXN];
int f[MAXN][Log+2];
int seq1[MAXN],seq2[MAXN],depth[MAXN],sum[MAXN];
int st[MAXN],ed[MAXN];
int n,x,y,cnt,q,m,w;

inline int lowbit(int x){return x&-x;}
inline void Add1(int x,int k){if(!x) return; while(x<=n) seq1[x]+=k,x+=lowbit(x);}
inline int Sum1(int x){int ans=0;while(x>0) ans+=seq1[x],x-=lowbit(x);return ans;}
inline void Add2(int x,int k){if(!x) return; while(x<=n) seq2[x]+=k,x+=lowbit(x);}
inline int Sum2(int x){int ans=0;while(x>0) ans+=seq2[x],x-=lowbit(x);return ans;}
/*
void Power(int k,int tot){
    while(k<=n+1){
        int kk=k;k<<=1;
        for(register int i=kk;i<=k-1;++i) s[i]=tot;
        ++tot;
    }
}
*/
void DFS(int fa,int rt){
    st[rt]=++cnt;f[rt][0]=fa;depth[rt]=depth[fa]+1;
    for(register int j=1;(1<<j)<=depth[rt];++j) f[rt][j]=f[f[rt][j-1]][j-1];
    for(register int i=0;i<tree[rt].size();++i) if(tree[rt][i]!=fa) DFS(rt,tree[rt][i]);
    ed[rt]=cnt;
}

inline int LCA(int x,int y){
    if(depth[x]<depth[y]) x^=y^=x^=y;
    for(register int k=/*s[depth[x]-depth[y]]*/19;k>=0;--k) if(depth[f[x][k]]>=depth[y]) x=f[x][k];
    if(x==y) return x;
    for(register int k=/*s[depth[x]-1]*/19;k>=0;--k) if(f[x][k]!=f[y][k]) x=f[x][k],y=f[y][k];
    return f[x][0];
}

int main(){//freopen("tmp.txt","r",stdin);
    n=Read(),m=Read();//root------1
    for(register int i=1;i<n;++i) x=Read(),y=Read(),tree[x].push_back(y),tree[y].push_back(x);
    DFS(0,1);//Power(1,0);
    for(register int i=1;i<=n;++i) sum[st[i]]=Read();
    for(register int i=1;i<=n;++i) sum[i]+=sum[i-1];
    for(register int i=1;i<=m;++i){
        q=Read();
        if(q==1){
            x=Read(),y=Read(),w=Read();
            int lca=LCA(x,y);
            Add1(st[x],w);Add1(st[y],w);Add1(st[lca],-w);Add1(st[f[lca][0]],-w);
            Add2(st[x],w*(depth[x]+1));Add2(st[y],w*(depth[y]+1));Add2(st[lca],-w*(depth[lca]+1));Add2(st[f[lca][0]],-w*(depth[f[lca][0]]+1));
        }
        else x=Read(),printf("%d
",sum[ed[x]]-sum[st[x]-1]+Sum2(ed[x])-Sum2(st[x]-1)-(Sum1(ed[x])-Sum1(st[x]-1))*depth[x]);
    }
    return 0;
}

4.对某个点X权值加上一个数W, 查询X到Y路径上所有点权之和

改点求链

用一个树状数组seq[i]seq[i]seq[i]表示在DFS序里的节点i到根节点的链权值和被加一个w,x到y的路径亦可拆成4条,改一个点以后就用差分思想把其子树点区间加w(树状数组区间修改seq,求点),查询时对x和y到根的链预处理好的权值和加上总操作值。可对比T2。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100000;
const int Log=19;
inline int Read(){
    int x=0,f=0;char ch;
    while(!isdigit(ch=getchar()))if(ch==‘-‘) f=1;
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^‘0‘),ch=getchar();
    return f?-x:x;
}
vector<int> tree[MAXN];
int f[MAXN][Log+2];
int len[MAXN],st[MAXN],ed[MAXN],A[MAXN],depth[MAXN],sum[MAXN];
int n,x,y,cnt,q,m,w;

inline int lowbit(int x){return x&-x;}
inline void Add(int x,int k){if(!x) return; while(x<=n) len[x]+=k,x+=lowbit(x);}
inline int Sum(int x){int ans=0;while(x>0) ans+=len[x],x-=lowbit(x);return ans;}
/*
void Power(int k,int tot){
    while(k<=n+1){
        int kk=k;k<<=1;
        for(register int i=kk;i<=k-1;++i) s[i]=tot;
        ++tot;
    }
}
*/
void DFS(int fa,int rt){
    st[rt]=++cnt;f[rt][0]=fa;depth[rt]=depth[fa]+1;sum[rt]=sum[fa]+A[rt];
    for(register int j=1;(1<<j)<=depth[rt];++j) f[rt][j]=f[f[rt][j-1]][j-1];
    for(register int i=0;i<tree[rt].size();++i) if(tree[rt][i]!=fa) DFS(rt,tree[rt][i]);
    ed[rt]=cnt;
}

inline int LCA(int x,int y){
    if(depth[x]<depth[y]) x^=y^=x^=y;
    for(register int k=/*s[depth[x]-depth[y]]*/19;k>=0;--k) if(depth[f[x][k]]>=depth[y]) x=f[x][k];
    if(x==y) return x;
    for(register int k=/*s[depth[x]-1]*/19;k>=0;--k) if(f[x][k]!=f[y][k]) x=f[x][k],y=f[y][k];
    return f[x][0];
}

int main(){//freopen("tmp.txt","r",stdin);
    n=Read(),m=Read();//root------1
    for(register int i=1;i<n;++i) x=Read(),y=Read(),tree[x].push_back(y),tree[y].push_back(x);
    for(register int i=1;i<=n;++i) A[i]=Read();
    DFS(0,1);//Power(1,0);
    for(register int i=1;i<=m;++i){
        q=Read();
        if(q==1) x=Read(),w=Read(),Add(st[x],w),Add(ed[x]+1,-w);
        else{
            x=Read(),y=Read();
            int lca=LCA(x,y);printf("%d
",sum[x]+sum[y]-sum[lca]-sum[f[lca][0]]+Sum(st[x])+Sum(st[y])-Sum(st[lca])-Sum(st[f[lca][0]]));
        }
    }
    return 0;
}

5.对子树X里所有节点加上一个值W, 查询某个点的值

改树求点

BIT+差分,水过。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100000;
inline int Read(){
    int x=0,f=0;char ch;
    while(!isdigit(ch=getchar()))if(ch==‘-‘) f=1;
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^‘0‘),ch=getchar();
    return f?-x:x;
}
vector<int> tree[MAXN];
int seq[MAXN],st[MAXN],ed[MAXN],A[MAXN];
int n,x,y,cnt,q,m;

inline int lowbit(int x){return x&-x;}
inline void Add(int x,int k){while(x<=n) seq[x]+=k,x+=lowbit(x);}
inline int Sum(int x){int ans=0;while(x>0) ans+=seq[x],x-=lowbit(x);return ans;}

void DFS(int fa,int rt){
    st[rt]=++cnt;
    for(register int i=0;i<tree[rt].size();++i) if(tree[rt][i]!=fa) DFS(rt,tree[rt][i]);
    ed[rt]=cnt;
}

int main(){
    n=Read(),m=Read();//root------1
    for(register int i=1;i<n;++i) x=Read(),y=Read(),tree[x].push_back(y),tree[y].push_back(x);
    DFS(0,1);
    for(register int i=1;i<=n;++i) A[i]=Read();
    for(register int i=1;i<=m;++i){
        q=Read();x=Read();
        if(q==1) y=Read(),Add(st[x],y),Add(ed[x]+1,-y);
        else printf("%d
",A[x]+Sum(st[x]));
    }
    return 0;
}

6.对子树X里所有节点加上一个值W, 查询某个子树的值

改树求树

线段树,区间修改区间查询水过。其实BIT也是可以的,看这里树状数组区间修改区间查询1或者这里树状数组区间修改区间查询2

#include<bits/stdc++.h>
#define l i<<1
#define r i<<1|1
#define mid ((L+R)>>1)
using namespace std;
const int MAXN=100000;
inline int Read(){
    int x=0,f=0;char ch;
    while(!isdigit(ch=getchar()))if(ch==‘-‘) f=1;
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^‘0‘),ch=getchar();
    return f?-x:x;
}
vector<int> tree[MAXN];
int st[MAXN],ed[MAXN],A[MAXN];
int sum[MAXN<<2],add[MAXN<<2];
int n,x,y,cnt,q,m,ql,qr,w;

void build(int i,int L,int R){
    if(L==R)sum[i]=A[L];
    else build(l,L,mid),build(r,mid+1,R),sum[i]=sum[l]+sum[r];
}

inline void pushdown(int i,int L,int R){
    if(add[i]) sum[l]+=(mid-L+1)*add[i],sum[r]+=(R-mid)*add[i],add[l]+=add[i],add[r]+=add[i],add[i]=0;
}

inline void Add(int i,int L,int R){
    if(ql<=L&&qr>=R){sum[i]+=w*(R-L+1),add[i]+=w;return;}
    pushdown(i,L,R);
    if(ql<=mid) Add(l,L,mid);if(qr>mid) Add(r,mid+1,R);
    sum[i]=sum[l]+sum[r];
}

inline int Query(int i,int L,int R){
    if(ql<=L&&qr>=R) return sum[i];
    pushdown(i,L,R);int ans=0;
    if(ql<=mid) ans+=Query(l,L,mid);if(qr>mid) ans+=Query(r,mid+1,R);
    return ans;
}

void DFS(int fa,int rt){
    st[rt]=++cnt;
    for(register int i=0;i<tree[rt].size();++i) if(tree[rt][i]!=fa) DFS(rt,tree[rt][i]);
    ed[rt]=cnt;
}

int main(){
    n=Read(),m=Read();//root------1
    for(register int i=1;i<n;++i) x=Read(),y=Read(),tree[x].push_back(y),tree[y].push_back(x);
    DFS(0,1);
    for(register int i=1;i<=n;++i) A[st[i]]=Read();
    build(1,1,n);
    for(register int i=1;i<=m;++i){
        q=Read(),x=Read();ql=st[x],qr=ed[x];
        if(q==1) w=Read(),Add(1,1,n);
        else printf("%d
",Query(1,1,n));
    }
    return 0;
}

7.对于子树X内的所有点都加上一个值W,查询X到Y之间的路径上所有点权和。

改树求链

写不动了。。

8.补充

改点求点

改点求链?

改链求链?

以上是关于复习:自己以前瞎写的树dfs序的主要内容,如果未能解决你的问题,请参考以下文章

瞎写的

强连通分量

BZOJ4009[HNOI2015]接水果 DFS序+整体二分+扫描线+树状数组

bzoj 4034(DFS序+线段树)

[leetcode] 100. 相同的树

[NOI2013]树的计数