[学习笔记]线段树合并

橙三吉。 提交于 2020-03-09 20:06:38

1、[POI2011]ROT-Tree Rotations

分析:线段树合并人生第一题。

网上的题解我都没看懂……我自己讲一下好了

线段树合并就是把两棵权值线段树合并到一棵

那怎么合并呢?

假设有这么两棵树:

一个结点代表一段值域区间有几个数,那么可以看出合并后应该是这样的

然后具体步骤就是找到一个结点,如果一个结点一棵树上有一棵树上没有,那么直
接返回那个结点的编号,否则两个值域的和相加,递归至左儿子和右儿子

每次合并的复杂度为两棵线段树的点数相加

这道题可以算出左儿子的逆序对个数,右儿子的逆序对个数,然后求出 \((x,y)\) 的对数, $x\in $ 左儿子, $y\in $ 右儿子

每次记录一下 \(sum\),在每次合并的时候 \(sum[lson[x]]\times sum[rson[y]]\)\(sum[rson[x]]\times sum[lson[y]]\) 中取个最小值算进答案的贡献就可以了(不懂可以自己画图或看代码理解一下)

\(Code\ Below:\)

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=200000+10;
int n,L[maxn*24],R[maxn*24],sum[maxn*24],cnt;
ll ans,num1,num2;

inline int read(){
    register int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return (f==1)?x:-x;
}

void update(int &now,int l,int r,int x){
    if(!now) now=++cnt;
    sum[now]++;
    if(l == r) return ;
    int mid=l+r>>1;
    if(x <= mid) update(L[now],l,mid,x);
    else update(R[now],mid+1,r,x);
}

void merge(int &x,int y){
    if(!x||!y){
        x=x+y;
        return ;
    }
    sum[x]+=sum[y];
    num1+=(ll)sum[R[x]]*sum[L[y]];
    num2+=(ll)sum[L[x]]*sum[R[y]];
    merge(L[x],L[y]);
    merge(R[x],R[y]);
}

void dfs(int &x){
    int val,lson=0,rson=0;
    val=read();
    if(!val){
        dfs(lson);dfs(rson);
        num1=num2=0;x=lson;
        merge(x,rson);
        ans+=min(num1,num2);
    }
    else update(x,1,n,val);
}

int main()
{
    n=read();
    int x=0;
    dfs(x);
    printf("%lld\n",ans);   
    return 0;
}

2、[Vani有约会]雨天的尾巴

分析:考虑树上查分。这次权值线段树维护最多救济粮的编号 \(val\) 和 最多救济粮的个数 \(sum\),在叶子节点更新一下,最后 \(pushup\) 一下

\(Code\ Below:\)

#include <bits/stdc++.h>
using namespace std;
const int maxn=100000+10;
const int size=100000;
int n,m,ans[maxn],rt[maxn],L[maxn*80],R[maxn*80],val[maxn*80],sum[maxn*80],cnt;
int head[maxn],to[maxn<<1],nxt[maxn<<1],tot;
int top[maxn],dep[maxn],siz[maxn],son[maxn],fa[maxn];

inline int read(){
    register int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return (f==1)?x:-x;
}
inline void add(int x,int y){
    to[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}

void dfs1(int x,int f){
    siz[x]=1;fa[x]=f;
    dep[x]=dep[f]+1;
    int maxson=-1;
    for(int i=head[x],y;i;i=nxt[i]){
        y=to[i];
        if(y==f) continue;
        dfs1(y,x);
        siz[x]+=siz[y];
        if(maxson<siz[y]){
            maxson=siz[y];
            son[x]=y;
        }
    }
}

void dfs2(int x,int topf){
    top[x]=topf;
    if(son[x]) dfs2(son[x],topf);
    for(int i=head[x],y;i;i=nxt[i]){
        y=to[i];
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}

int LCA(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    return x;
}

void pushup(int now){
    if(sum[L[now]]>sum[R[now]]) sum[now]=sum[L[now]],val[now]=val[L[now]];
    else if(sum[L[now]]<sum[R[now]]) sum[now]=sum[R[now]],val[now]=val[R[now]];
    else sum[now]=sum[L[now]],val[now]=min(val[L[now]],val[R[now]]);
}

void update(int &now,int l,int r,int x,int v){
    if(!now) now=++cnt;
    if(l == r){sum[now]+=v;val[now]=sum[now]?l:0;return ;}
    int mid=l+r>>1;
    if(x<=mid) update(L[now],l,mid,x,v);
    else update(R[now],mid+1,r,x,v);
    pushup(now);
}

void merge(int &x,int y,int l,int r){
    if(!x||!y){x=x+y;return ;}
    if(l == r){sum[x]+=sum[y];val[x]=sum[x]?l:0;return ;}
    int mid=l+r>>1;
    merge(L[x],L[y],l,mid);
    merge(R[x],R[y],mid+1,r);
    pushup(x);
}

void dfs(int x,int f){
    for(int i=head[x],y;i;i=nxt[i]){
        y=to[i];
        if(y==f) continue;
        dfs(y,x);
        merge(rt[x],rt[y],1,size);
    }
    ans[x]=val[rt[x]];
}

int main()
{
    n=read(),m=read();
    int x,y,z,lca;
    for(int i=1;i<n;i++){
        x=read(),y=read();
        add(x,y);add(y,x);
    }
    dfs1(1,0);dfs2(1,1);
    for(int i=1;i<=m;i++){
        x=read(),y=read(),z=read();
        lca=LCA(x,y);
        update(rt[x],1,size,z,1);
        update(rt[y],1,size,z,1);
        update(rt[lca],1,size,z,-1);
        if(fa[lca]) update(rt[fa[lca]],1,size,z,-1);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++)
        printf("%d\n",ans[i]);
    return 0;
}

3、CF600E Lomsat gelral

分析:这次再维护一个 \(val\) 最大的和 \(ans\),从下向上合并即可

\(Code\ Below:\)

#include <bits/stdc++.h>
#define mid (l+r>>1)
#define lson L[now]
#define rson R[now]
#define int long long
using namespace std;
const int maxn=100000+10;
const int size=100000;
int n,m,a[maxn],an[maxn],rt[maxn],cnt;
int L[maxn*20],R[maxn*20],sum[maxn*20],val[maxn*20],ans[maxn*20];
int head[maxn],to[maxn<<1],nxt[maxn<<1],tot;

inline int read(){
    register int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return (f==1)?x:-x;
}
inline void add(int x,int y){
    to[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}

void pushup(int now){
    sum[now]=(sum[lson]>sum[rson])?sum[lson]:sum[rson];
    val[now]=(sum[lson]>sum[rson])?val[lson]:val[rson];
    ans[now]=((sum[lson]>=sum[rson])?ans[lson]:0)+((sum[lson]<=sum[rson])?ans[rson]:0);
}

void update(int &now,int l,int r,int x){
    if(!now) now=++cnt;
    if(l == r){sum[now]++;val[now]=ans[now]=l;return ;}
    if(x <= mid) update(lson,l,mid,x);
    else update(rson,mid+1,r,x);
    pushup(now);
}

void merge(int &x,int y,int l,int r){
    if(!x||!y){x=x+y;return ;}
    if(l == r){sum[x]+=sum[y];val[x]=ans[x]=l;return ;}
    merge(L[x],L[y],l,mid);
    merge(R[x],R[y],mid+1,r);
    pushup(x);
}

void dfs(int x,int f){
    for(int i=head[x],y;i;i=nxt[i]){
        y=to[i];
        if(y==f) continue;
        dfs(y,x);
        merge(rt[x],rt[y],1,size);
    }
    an[x]=ans[rt[x]];
}

signed main()
{
    n=read();
    int x,y;
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1;i<n;i++){
        x=read(),y=read();
        add(x,y);add(y,x);
    }
    for(int i=1;i<=n;i++)
        update(rt[i],1,size,a[i]);
    dfs(1,0);
    for(int i=1;i<=n;i++)
        printf("%lld ",an[i]);
    printf("\n");
    return 0;
}

搞了一个下午线段树合并,还是很有成就感滴!

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!