洛谷4374 [USACO18OPEN]Disruption（树链剖分）|（LCA+并查集）

题目

洛谷4374 [USACO18OPEN]Disruption

题解１:在线

树链剖分+线段树维护最小值
一条新路(x,y,c)影响的是树上(x,y)间的所有路径，要取min，很容易想到用线段树维护。

题解２：离线

排序+并查集
可以把所有新路递增排序，所以前面更新过的路后面一定不会再更新。
暴力的话就是把(x,lca)和(y,lca)的路径全部走一遍。一种巧妙的方法是用并查集做路径压缩。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
const int MAXN=50010;

int n,m;

struct E{int x,y,c,next;}e[MAXN*2],a[MAXN];int len=1,last[MAXN],tk[MAXN];
bool cmp(E e1,E e2){return e1.c<e2.c;}
void ins(int x,int y)
{
    e[++len]=(E){x,y,INF,last[x]};last[x]=len;
}

int head,tail,q[MAXN];
int dep[MAXN],f[MAXN][20];
void bfs()
{
    dep[1]=1;
    head=0,tail=1;q[head]=1;
    while(head<tail)
    {
        int x=q[head++];
        for(int k=last[x];k;k=e[k].next)
        {
            int y=e[k].y;
            if(dep[y]) continue;
            tk[y]=k;
            dep[y]=dep[x]+1;
            f[y][0]=x;for(int i=1;i<=16;i++) f[y][i]=f[f[y][i-1]][i-1];
            q[tail++]=y;
        }
    }
}
int lca(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=16;i>=0;i--)
        if(dep[f[x][i]]>=dep[y]) x=f[x][i];
    if(x==y) return x;
    for(int i=16;i>=0;i--)
        if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}

int fa[MAXN];
int findfa(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=findfa(fa[x]);
}
void change(int x,int p,int c)
{
    while(dep[x]>dep[p])
    {
        e[tk[x]].c=e[tk[x]^1].c=min(e[tk[x]].c,c);
        fa[x]=f[x][0];
        x=findfa(x);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    bfs();
    for(int i=1;i<=m;i++) scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].c);
    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x=a[i].x,y=a[i].y,p=lca(x,y);
        if(x!=p) change(x,p,a[i].c);
        if(y!=p) change(y,p,a[i].c);
    }
    for(int i=2;i<=len;i+=2)
        if(e[i].c==INF) puts("-1");
        else printf("%d\n",e[i].c);
    return 0;
}