P3258 [JLOI2014]松鼠的新家(树上差分板子)

树上差分!!!

这是一道模板题的稍微改编

首先,啥是树上差分

就和名字一样,书上的差分

倘若我们要给(u,v)的路径上加上k

则给u+k,v+k,然后lca(u,v)-k(因为事实上lca处就加了一个,可是这么干加了俩) father(lca(---))-k(这里及更高处根本没动过,消除影响)

所以这题就是lca+树上差分

但是和模板题的不同是,这里,每一点(除了一开始的和重点),即使起始点也是终点,而且最后一点不用放。

需要特别处理

      #include<iostream>
#include<cstdio>
using namespace std;
int head[2*500001];
int p;
struct b{
    int to;
    int ne;
} e[2*500001];
int fa[2*500001][30];
int dep[2*500001];
int n,m,s;
int logg[2*500001];
int ma[300005]; 
int cha[300005];
int ans[300005];
int cnt[300005];
int x,y;
void dfs(int now,int fat){
    dep[now]=dep[fat]+1;
    fa[now][0]=fat;
    for(int i=1;(1<<i)<=dep[now];++i){
        fa[now][i]=fa[fa[now][i-1]][i-1];
    }
    for(int i=head[now];i;i=e[i].ne){
        if(e[i].to!=fat)
        dfs(e[i].to,now);
    }
}
int lca(int x,int y){
    if(dep[x]<dep[y]){
        swap(x,y);
    }
    while(dep[x]>dep[y])
    x=fa[x][logg[dep[x]-dep[y]]-1];
    if(x==y)
    return x;
    for(int k=logg[dep[x]]-1;k>=0;k--){
        if(fa[x][k]!=fa[y][k]){
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    return fa[x][0];
}
void print(int x,int fa){
//  cout<<x<<endl;
    for(int i=head[x];i;i=e[i].ne){
//          cout<<e[i].to;
        if(e[i].to!=fa){
            print(e[i].to,x);
            ans[x]+=ans[e[i].to];
        }
    }
    ans[x]+=cha[x];
    return ;
}
void add(int f,int t){
    p++;
    e[p].to=t;
    e[p].ne=head[f];
    head[f]=p;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&ma[i]);
        if(i!=1)
        cnt[ma[i]]++;
        }
    for(int i=1;i<=n-1;++i){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    for(int i=1;i<=n;++i)
        logg[i]=logg[i-1]+(1<<logg[i-1]==i);
    for(int i=1;i<n;++i){
        int o=lca(ma[i],ma[i+1]);
        int v=ma[i+1];
        cha[ma[i]]++;
        cha[v]++;
        cha[o]--;
        cha[fa[o][0]]--;
    }
    print(1,0);
    for(int i=1;i<=n;++i){
        ans[i]-=cnt[i];
        cout<<ans[i]<<endl;
    }
    return 0;
}
暂无评论

发送评论 编辑评论


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
上一篇
下一篇