求该做法时间复杂度
查看原帖
求该做法时间复杂度
455757
yydfj楼主2023/10/12 19:53

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 和树链剖分,而且跑得还巨快,拿下了最优解,想问一下这种做法的时间复杂度。

2023/10/12 19:53
加载中...