差分[差分数组 & 树状差分]

Posted hzoi-poozhai

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了差分[差分数组 & 树状差分]相关的知识,希望对你有一定的参考价值。

差分[差分数组 & 树状差分]

  1. 差分数组

  • 差分数组的定义:记录当前位置的数与上一位置的数的差值.
原数组 ai 9 4 7 5 9
差分数组 bi 9 -5 3 -2 4
差分数组的前缀和 9 4 7 5 9

显然通过求前缀和可以做到单点查询

他高效的地方在于区间修改,比如我们对区间[2,4]每个元素加上5,我们只需在差分数组:b2+=5,b5?=5,然后求前缀和即可

原数组 ai 9 4 7 5 9
差分数组 bi 9 0 3 -2 -1
差分数组的前缀和 9 9 12 10 9

[1,2]仍不变, [2,4]的前缀和都加了5, 4以后的话+5,-5相抵消, 还是不变的

因此差分数组能够高效的解决区间修改单点查询的问题。

  1. 树状差分

树的以下两个性质:

  1. 任意两个节点之间有且只有一条路径。
  2. 根节点确定时,一个节点只有一个父亲节点。

如果假设我们要考虑的是从uv的路径,uvlcaa,,我们将路径拆分成 u->aa->v

如果题目要求对树上的一段路径进行操作,并询问某个点或某条边被经过的次数,树上差分就可以派上用场了。

点差分

技术图片

设原数组为a,差分数组为d。假如给d[i]+1,其实就相当于给a[i]~a[n]每个元素+1

如果给树上的一个节点xd[x] +1, 相当于a[root]~a[x]链上的每个节点都+1

假设a=lca(x,y)。 把链x~y分成两个链,x~aa~y。即d[x]+1,d[y]+1

但这样链a~root增加了2,我们让d[a]-1,此时fa[a]~root变成了-1,需要d[fa[a]]-1完美解决。

例题:松鼠的新家(luogu p3258)

Description

  • 松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有 n-1 根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的。天哪,他居然真的住在“树”上。
  • 松鼠想邀请小熊前来参观,并且还指定一份参观指南,他希望小熊能够按照他的指南顺序,先去 a1a1,再去 a2a2,……,最后到 anan,去参观新家。可是这样会导致重复走很多房间,懒惰的维尼不停地推辞。可是松鼠告诉他,每走到一个房间,他就可以从房间拿一块糖果吃。
  • 小熊是个馋家伙,立马就答应了。现在松鼠希望知道为了保证维尼有糖果吃,他需要在每一个房间各放至少多少个糖果。
  • 因为松鼠参观指南上的最后一个房间 anan 是餐厅,餐厅里他准备了丰盛的大餐,所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。

Input

  • 第一行一个正整数 n,表示房间个数
  • 第二行 n个正整数,依次描述 a1,a2,?,ana1,a2,?,an。
  • 接下来 n-1 行,每行两个正整数 x,y,表示标号 xy的两个房间之间有树枝相连。

Output

  • 一共 n 行,第 i 行输出标号为 i 的房间至少需要放多少个糖果,才能让小熊有糖果吃。

Input

5
1 4 5 3 2
1 2
2 4
2 3
4 5

output

1
2
1
2
1    
#include <bits/stdc++.h>
using namespace std;
const int maxn=3e5+5;
struct edge{ int to,next; }e[2*maxn];
int n, len, a[maxn], head[maxn], dep[maxn], f[maxn][21], sum[maxn];
void Insert(int u, int v){ e[++len].to=v;e[len].next=head[u];head[u]=len; }
void dfs(int u, int fa){//dfs预处理祖先, 不用多说吧
    dep[u] = dep[fa]+1; f[u][0] = fa;
    for(int i=1; (1<<i)<=dep[u]; i++) f[u][i] = f[f[u][i-1]][i-1];
    for(int i=head[u]; i; i=e[i].next){
        int v = e[i].to;
        if(v != fa) dfs(v, u);
    }
}
int lca(int u, int v){//获取lca
    if(dep[u] < dep[v]) swap(u, v);
    int len = dep[u]-dep[v], k = 0;
    while(len){
        if(len & 1)u=f[u][k];
        ++k; len>>=1;
    }    
    if(u==v) return u;
    for(int i=20; i>=0; i--) if(f[u][i] != f[v][i]) u = f[u][i], v = f[v][i];            
    return f[u][0];
}
void search(int u){//求每个节点的差分和
    for(int i=head[u]; i; i=e[i].next){
        int v = e[i].to;
        if(v == f[u][0]) continue;
        search(v);
        sum[u] += sum[v];
    }
}
int main(){
    scanf("%d", &n);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]);
    for(int i=1; i<=n-1; i++){
        int u, v; scanf("%d%d", &u, &v);
        Insert(u, v); Insert(v, u);
    }
    dfs(1,0);
    for(int i=1; i<=n-1; i++){
        int x=a[i], y=a[i+1], LCA=lca(x,y);        
        sum[x]++;
        sum[y]++;
        sum[LCA]--;
        sum[f[LCA][0]]--;
    }
    search(1);
    for(int i=2; i<=n; i++) sum[a[i]]--;//2-n的点都多算了一次
    for(int i=1; i<=n; i++) printf("%d
",sum[i]);
    return 0;
}

边差分

cf[i]代表从ii的父亲这一条路径经过的次数。令a=lca(u,v) 因为关于边的差分,a表示a到其父亲的那条边,不在u~v路径中,所以cf[u]++,cf[v]++,cf[a]?=2

  • 例题:luogu P2680 运输计划
  • 题意:一棵树上有m条道路,可以使任意一条道路的权值变为0,怎样使长度最长的道路长度最小。

思路 最小的长度, 肯定在 0 - maxl(最长的那条道路) 之间, 我们可以用二分在 0-maxl枚举, 每次枚举统计长度超过mid的路径的条数cnt, 找到一条被经过了cnt次的边(这条边可以影响所有长度超过mid的路径), 把符合条件的边中最长的那一条归零, 如果cnt条路径中最长的减去这条边权<=mid, 则当前的mid合法.

#include <bits/stdc++.h>
using namespace std;
const int maxn=3e5+5;
int f[maxn][22],head[maxn],edge[maxn],dep[maxn],dis[maxn];
int st[maxn],ed[maxn],c[maxn],lca[maxn],len[maxn];
int n,m,lene,maxl,ans,cnt,max_edge;
struct Edge{int to,w,next;}e[maxn<<1];
void Insert(int x,int y,int z){e[++lene].to=y; e[lene].w=z; e[lene].next=head[x]; head[x]=lene;}
void dfs(int u){//dfs预处理祖先
    dep[u] = dep[f[u][0]] + 1;
    for(int i=1; (1<<i)<=dep[u]; i++) f[u][i]=f[f[u][i-1]][i-1];
    for(int i=head[u]; i; i=e[i].next){
        int v=e[i].to;
        if(v==f[u][0])continue;
        f[v][0]=u;
        edge[v]=e[i].w;//edge[v]记录v到父节点u的边长
        dis[v]=dis[u]+e[i].w;//求解v到根节点的距离
        dfs(v);
    }
}
int Lca(int u,int v){
    if(dep[u] < dep[v]) swap(u, v);
    int len = dep[u]-dep[v], k = 0;
    while(len){
        if(len & 1) u = f[u][k];
        ++k; len >>= 1;
    }
    if(u==v) return u;
    for(int i=20; i>=0; i--) if(f[u][i] != f[v][i]) u = f[u][i], v = f[v][i];
    return f[u][0];
}
int getdis(int i){ return dis[st[i]] + dis[ed[i]] - 2*dis[lca[i]]; } //求第i条路径长度
int dfs2(int u){
    int tot = c[u];//计算u到其父亲这条边访问次数
    for(int i=head[u]; i; i=e[i].next){
        int v = e[i].to;
        if(v==f[u][0])continue;
        tot += dfs2(v);
    }
    if(tot==cnt) max_edge = max(max_edge, edge[u]);
    return tot;
}
bool check(int mid){
    cnt=0, max_edge=0;//cnt记录超过mid的路径数,max_edge记录要删掉的边
    for(int i=1; i<=n; i++) c[i] = 0;//差分数组清零 
    for(int i=1; i<=m; i++)
        if(len[i] > mid)//超出mid的区间进行差分,并计数
            cnt++, c[st[i]]++, c[ed[i]]++, c[lca[i]]-=2;
    if(cnt==0)return 1;//没有超过的
    dfs2(1);//求出被经过cnt次,且最长的边
    return maxl-max_edge <= mid;
}
int main(){
    scanf("%d%d", &n, &m);
    for(int i=1; i<n; i++){
        int u, v, w; scanf("%d%d%d", &u, &v, &w);
        Insert(u, v, w); Insert(v, u, w);
    }
    dfs(1);
    for(int i=1; i<=m; i++){
        scanf("%d%d", &st[i], &ed[i]);
        lca[i] = Lca(st[i], ed[i]);
        len[i] = getdis(i);
        maxl = max(maxl,len[i]);
    }
    int l = 0, r = maxl;//最短时间在[0,maxl]之间
    while(l <= r){
        int mid = (l+r)>>1;
        if(check(mid)) ans = mid, r = mid-1;
        else l = mid+1;
    }
    printf("%d
",ans);
    return 0;
}

以上是关于差分[差分数组 & 树状差分]的主要内容,如果未能解决你的问题,请参考以下文章

未解决的树状数组差分(改变区间,求单点值)

luogu P4514ybtoj树状数组课堂过关差分 例题6区间修改区间查询 & 上帝造题的七分钟

树状数组差分求前缀和的前缀和

树状数组区间修改,区间更新:差分数组的运用

ybtoj树状数组差分例题4区间修改区间查询

P3368模板树状数组 2 - 差分