题目
洛谷4374 [USACO18OPEN]Disruption
题解1:在线
树链剖分+线段树维护最小值
一条新路(x,y,c)影响的是树上(x,y)间的所有路径,要取min,很容易想到用线段树维护。
题解2:离线
排序+并查集
可以把所有新路递增排序,所以前面更新过的路后面一定不会再更新。
暴力的话就是把(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;
}