Rt,先上代码:
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,nxt[100005],to[100005],val[100005],h[50005],cnt;
int id[50005],p[50005],f[50005],dep[50005],ans[50005];
struct node{
int a,b,c;
}rd[50005];
bool cmp(node x,node y){return x.c<y.c;}
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
return x*f;
}
void add(int x,int y,int z)
{
nxt[++cnt]=h[x];
to[cnt]=y;
val[cnt]=z;
h[x]=cnt;
}
void dfs(int x,int y)
{
p[x]=y;
dep[x]=dep[y]+1;
for(register int i=h[x];i;i=nxt[i])
if(to[i]!=y)
{
id[to[i]]=val[i];
dfs(to[i],x);
}
}
inline int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
int main()
{
n=read();m=read();
for(register int i=1;i<n;i++)
{
int x,y;
x=read();y=read();
add(x,y,i);add(y,x,i);
ans[i]=1e9;
f[i]=i;
}
f[n]=n;
dfs(1,0);
for(register int i=1;i<=m;i++) rd[i].a=read(),rd[i].b=read(),rd[i].c=read();
stable_sort(rd+1,rd+m+1,cmp);
for(register int i=1;i<=m;i++)
for(register int a=find(rd[i].a),b=find(rd[i].b);a!=b;)
{
if(dep[a]<dep[b]) swap(a,b);
f[a]=f[p[a]];
ans[id[a]]=rd[i].c;
a=find(a);b=find(b);
}
for(register int i=1;i<n;i++)
if(ans[i]==1e9) printf("-1\n");
else printf("%d\n",ans[i]);
return 0;
}
只用了并查集就过掉了,并没有用 LCA 和树链剖分,而且跑得还巨快,拿下了最优解,想问一下这种做法的时间复杂度。