Codeforces 1184 E3 Daleks' Invasion (hard) —— 树链剖分,替换最小生成树

This way

题意:

给你一张图,n个点,m条边,问你每条边可以替换成的最大值(<=1e9),使得替换后能在最小生成树上。

题解:

在这里插入图片描述
这些都是什么神仙做法,而且他们的代码长度也短,更重要的是百度我还找不到。
那我这种弱鸡只能想到树链剖分了,而且代码贼长。
首先将边分成原本就在树上的情况和不在树上的情况。那么首先对于不在树上的情况很好考虑,一定是替换成两个节点路径上的最大值。
那么对于在树上的情况呢,如果将这条边去掉,那么是从它的子树上选一条最短的边连向除了它子树部分的点,那么如果dfs去做我感觉很难做,因为不好维护下面传上来的边,然后就发现了,其实下面的边也有一个存在的范围,也就是两个点之间的路径。
但是要注意我们要将存边改为存点,也就是将边权变成点权,存到儿子节点上面。那么此时就需要求出两个点的lca,然后分成两个线段去存,查询也是一样。
代码极长无比,难怪只有这么点人去做。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pa pair<int,int>
const int N=1e6+5;
struct node
{
    int to,next;
}e[N*2];
int head[N],cnt,tim,fa[N][21],son[N],dep[N],top[N],en[N],siz[N],pos[N];
int f_mx[N*4],mx[N*4],mi[N*4],f_mi[N*4];
int n,m,rt;
void add(int x,int y)
{
    e[cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt++;
}
void dfs1(int u)
{
    siz[u]=1;
    for (int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if (v!=fa[u][0])
        {
            fa[v][0]=u;
            dep[v]=dep[u]+1;
            dfs1(v);
            siz[u]+=siz[v];
            if (son[u]==0||siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}
void dfs2(int u,int f)
{
    tim++;
    top[u]=f;
    pos[u]=tim;
    if (son[u]) dfs2(son[u],f);
    for (int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].to;
        if (v!=fa[u][0]&&v!=son[u])
        {
            dfs2(v,v);
        }
    }
    en[u]=tim;
}
void pushdown(int root)
{
    mx[root<<1]=max(mx[root<<1],f_mx[root]);
    mx[root<<1|1]=max(mx[root<<1|1],f_mx[root]);
    f_mx[root<<1]=max(f_mx[root<<1],f_mx[root]);
    f_mx[root<<1|1]=max(f_mx[root<<1|1],f_mx[root]);
    f_mx[root]=0;
    mi[root<<1]=min(mi[root<<1],f_mi[root]);
    mi[root<<1|1]=min(mi[root<<1|1],f_mi[root]);
    f_mi[root<<1]=min(f_mi[root<<1],f_mi[root]);
    f_mi[root<<1|1]=min(f_mi[root<<1|1],f_mi[root]);
    f_mi[root]=1e9;
}
void add(int l,int r,int root,int ql,int qr,int val,int op)
{
    if(l>=ql&&r<=qr)
    {
        if(op){
            mx[root]=max(mx[root],val);
            f_mx[root]=max(f_mx[root],val);
        }
        else{
            mi[root]=min(mi[root],val);
            f_mi[root]=min(f_mi[root],val);
        }
        return ;
    }
    pushdown(root);
    int mid=l+r>>1;
    if(mid>=ql)
        add(l,mid,root<<1,ql,qr,val,op);
    if(mid<qr)
        add(mid+1,r,root<<1|1,ql,qr,val,op);
    mx[root]=max(mx[root<<1],mx[root<<1|1]);
    mi[root]=min(mi[root<<1],mi[root<<1|1]);
}
int q_mx(int l,int r,int root,int ql,int qr)
{
    if(l>=ql&&r<=qr)
        return mx[root];
    int mid=l+r>>1;
    pushdown(root);
    int ans=0;
    if(mid>=ql)
        ans=q_mx(l,mid,root<<1,ql,qr);
    if(mid<qr)
        ans=max(ans,q_mx(mid+1,r,root<<1|1,ql,qr));
    return ans;
}
int q_mi(int l,int r,int root,int p){
    if(l==r)
        return mi[root];
    int mid=l+r>>1;
    pushdown(root);
    if(mid>=p)
        return q_mi(l,mid,root<<1,p);
    else
        return q_mi(mid+1,r,root<<1|1,p);
}
void update(int x,int y,int val,int op)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        add(1,n,1,pos[top[x]],pos[x],val,op);
        x=fa[top[x]][0];
    }
    if (pos[x]>pos[y]) swap(x,y);
    add(1,n,1,pos[x],pos[y],val,op);
}
int query(int x,int y)
{
    int ans=0;
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=max(ans,q_mx(1,n,1,pos[top[x]],pos[x]));
        x=fa[top[x]][0];
    }
    if (pos[x]>pos[y]) swap(x,y);
    ans=max(ans,q_mx(1,n,1,pos[x],pos[y]));
    return ans;
}
struct edge{
    int x,y,v,id;
    bool operator<(const edge& a)const
    {
        return v<a.v;
    }
}p[N];
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    for(int i=20;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            x=fa[x][i];
    if(x==y)return x;
    for(int i=20;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int f[N],in[N];
int finds(int x){return x==f[x]?f[x]:f[x]=finds(f[x]);}
void deal(){
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
}
int ans[N];
int main()
{
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].v),p[i].id=i;
    dep[1]=1;
    sort(p+1,p+1+m);
    for(int i=1;i<=m;i++){
        int fax=finds(p[i].x);
        int fay=finds(p[i].y);
        if(fax!=fay){
            in[p[i].id]=1;
            f[fay]=fax;
            add(p[i].x,p[i].y);
            add(p[i].y,p[i].x);
        }
    }
    dfs1(1);
    dfs2(1,1);
    deal();
    for(int i=1;i<=n*4;i++)
        mi[i]=f_mi[i]=1e9;
    for(int i=1;i<=m;i++){//分情况更新,如果在树上就更新最大值,不在树上就更新最小值
        int x=p[i].x,y=p[i].y,firstson,val=p[i].v;
        int l=lca(x,y);
        if(y!=l)
        {
            firstson=y;
            for(int i=20;i>=0;i--)
                if(dep[fa[firstson][i]]>dep[l])
                    firstson=fa[firstson][i];
            update(firstson,y,val,in[p[i].id]?1:0);
        }
        if(x!=l)
        {
            firstson=x;
            for(int i=20;i>=0;i--)
                if(dep[fa[firstson][i]]>dep[l])
                    firstson=fa[firstson][i];
            update(firstson,x,val,in[p[i].id]?1:0);
        }
    }
    for(int i=1;i<=m;i++){
        if(in[p[i].id])//如果在树上就查询儿子节点的最小值
            ans[p[i].id]=q_mi(1,n,1,dep[p[i].x]>dep[p[i].y]?pos[p[i].x]:pos[p[i].y]);
        else{//否则就查询最大值
            int x=p[i].x,y=p[i].y,firstson,val=p[i].v;
            int l=lca(x,y);
            int aa=0;
            if(y!=l)
            {
                firstson=y;
                for(int i=20;i>=0;i--)
                    if(dep[fa[firstson][i]]>dep[l])
                        firstson=fa[firstson][i];
                aa=query(firstson,y);
            }
            if(x!=l)
            {
                firstson=x;
                for(int i=20;i>=0;i--)
                    if(dep[fa[firstson][i]]>dep[l])
                        firstson=fa[firstson][i];
                aa=max(aa,query(firstson,x));
            }
            ans[p[i].id]=aa;
        }
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值