MLE求助
查看原帖
MLE求助
401393
一只绝帆楼主2021/12/26 16:58

不理解为什么MLE,求助大佬们

#include<bits/stdc++.h>
using namespace std;
const int N=2e4+5,C=1e5+5;
int n,m,s,u[N],v[N],start[N],Next[N],f[N][20],g[N][20],d[N],cnt,f1[N],q,vis[N];
struct B{
	int u;
	int v;
	int w;
} bi[C];
int find(int x)
{
	if(x==f1[x])
		return x;
	else 
		return f1[x]=find(f1[x]);
}
bool cmp(B x,B y)
{
	return x.w>y.w;
}
void dfs_f(int now)
{
	vis[now]=1;
	d[now]=d[f[now][0]]+1;
	for(int i=start[now];i!=0;i=Next[i])
	{
		if(bi[i].v==f[now][0])
		{
			continue;
		 } 
		f[bi[i].v][0]=now;
		g[bi[i].v][0]=bi[i].w;
		dfs_f(bi[i].v);
	}
}
int lca_g(int x,int y)
{
	if(find(x)!=find(y)) return -1;
	int ans=0x7f7f7f7f;
	if(d[y]>d[x]) swap(x,y);
	int k=log2(d[x]);
	for(int i=k;i>=0;i--)
	{
		if(d[f[x][i]]>=d[y]) ans=min(ans,g[x][i]),x=f[x][i];
	}
	if(x==y) return ans;
	for(int i=k;i>=0;i--)
	{
		if(f[x][i]==f[y][i]) continue;
		ans=min(ans,g[x][i]);
		ans=min(ans,g[y][i]);
		x=f[x][i];
		y=f[y][i];
	}
	return min(ans,min(g[x][0],g[y][0]));
}
int main(){
	memset(g,0x7f,sizeof g);
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		cin>>bi[i].u>>bi[i].v>>bi[i].w;
	}
	for(int i=1;i<=n;i++)
	{
		f1[i]=i;
	}
	sort(bi+1,bi+m+1,cmp);
	for(int i=1;i<=m;i++)
	{
		int fu=find(bi[i].u),fv=find(bi[i].v);
		if(fu==fv)
		{
			continue;
		}
		cnt++;
		f1[fu]=fv;
		bi[i+m].u=bi[i].v;
		bi[i+m].v=bi[i].u;
		bi[i+m].w=bi[i].w;
		Next[i]=start[bi[i].u];
		start[bi[i].u]=i;
		Next[i+m]=start[bi[i+m].u];
		start[bi[i+m].u]=i+n;
		if(cnt==n-1) 
		{
			break;
		}
	}
	
	for(int i=1;i<=n;i++)
	{
		if(!vis[i]) dfs_f(i);
	}
	g[0][0]=0x7f7f7f7f;//叶子节点会把此点赋成0 
	for(int j=1;(1<<j)<=n;j++)
	{
		for(int i=1;i<=n;i++)
		{
			f[i][j]=f[f[i][j-1]][j-1];
			g[i][j]=min(g[i][j-1],g[f[i][j-1]][j-1]);
		}
	}
	cin>>q;
	for(int i=1;i<=q;i++)
	{
		int x,y;
		cin>>x>>y;
		cout<<lca_g(x,y)<<endl;
		
	}
	return 0;
}
2021/12/26 16:58
加载中...