MLE了,求救
查看原帖
MLE了,求救
667000
manstain楼主2024/4/3 18:22

有大佬能帮蒟蒻本弱抢救一下吗

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <utility>
#include <unordered_set>


const int N=100000;
int n,m,q;
int head[N+5],from[6*N+5],to[6*N+5],next[6*N+5],W[6*N+5];
int cnt=-1;
int p[N+5];
int f[N+5][20],sta[N+5][20],depths[N+5];
bool vis[N+5];
std::pair<int,int> W1[3*N+5];
int p1[N+5];
bool used[N+5];



void add(int u,int v,int w){
	next[++cnt]=head[u];
	head[u]=cnt;
	to[cnt]=v;
	from[cnt]=u;
	W[cnt]=w;
}

void del(int r){
	int u=from[r];
    int last=0;
	for(int i=head[u];~i;i=next[i]){
		if(r==i){
	       if(i==head[u]){
		   	  head[u]=next[i];
		   	  break;
		   }
		   next[last]=next[i];
		   break;
		}
		last=i;
	}
}
void init(){
	for(int i=1;i<=n;i++) p[i]=i;
}
int find(int x){
	if(p[x]==x) return x;
	else{
		p[x]=find(p[x]);
		return p[x];
	}
	
}

void unionn(int a,int b){
	int pa=find(a);
	int pb=find(b);
	if(pa!=pb) p[p[b]]=p[a];
}

void init1(){
	for(int i=1;i<=n;i++) p1[i]=i;
}
int find1(int x){
	if(p1[x]==x) return x;
	else{
		p1[x]=find1(p1[x]);
		return p1[x];
	}
	
}

void unionn1(int a,int b){
     p1[p1[b]]=p1[a];
}

void dfs(int u,int fa,int begin){
		for(int i=head[u];~i;i=next[i]){
		int v=to[i];
		if(v!=fa&&v!=begin){
			unionn(u,v);
			vis[v]=1;
			dfs(v,u,begin);
		}
	}
}

void kruscal(){
	int sum=(cnt+1)>>1;
	init1();
	std::sort(W1,W1+sum);
	for(int i=sum-1;i>=0;i--){
		int u=from[W1[i].second],v=to[W1[i].second];
        if(find1(u)==find1(v)){
		del(W1[i].second);
		del(W1[i].second+1);
		}
		else{
			unionn1(u,v);
		}
	}
}

void dfs1(int u,int fa){
	for(int i=head[u];~i;i=next[i]){
		int v=to[i];
		if(v!=fa){
			depths[v]=depths[u]+1;
			f[v][0]=u;
			sta[v][0]=W[i];
		    for(int j=1;(1<<j)<=depths[v];j++){
				f[v][j]=f[f[v][j-1]][j-1];
				sta[v][j]=std::min(sta[v][j-1],sta[f[v][j-1]][j-1]);
			}
			dfs1(v,u);
		}
	}
}

int lca(int u,int v){
	int ans=1000005;
	if(depths[u]>depths[v]) std::swap(u,v);
	for(int i=18;i>=0;i--){
		if((1<<i)>depths[v]-depths[u]) continue;
		ans=std::min(ans,sta[v][i]);
		v=f[v][i];
	}
	if(u==v) return ans;
	for(int i=18;i>=0;i--){
		if(f[v][i]==f[u][i]) continue;
		ans=std::min(sta[u][i],ans);
		ans=std::min(sta[v][i],ans);
		u=f[u][i],v=f[v][i];
	}
	ans=std::min(ans,sta[u][0]);
	ans=std::min(ans,sta[v][0]);
	return ans;
} 

int main(){
	memset(head,-1,sizeof(head));
	scanf("%d %d %d",&n,&m,&q);
	for(int i=1;i<=m;i++){
		int u,v,w;
		scanf("%d %d %d",&u,&v,&w);
		add(u,v,w);
		W1[cnt>>1]={w,cnt};
		add(v,u,w);
	}
	init();
    vis[1]=1;
	dfs(1,0,1); 
	for(int i=2;i<=n;i++){
	    if(!vis[i]) {
	    	vis[i]=1;
			dfs(i,0,i);
		}
	}
	for(int i=1;i<=n;i++) find(i);
	kruscal();
	for(int i=1;i<=n;i++){
		if(!used[p[i]]){
			used[p[i]]=1;
			dfs1(p[i],0);
		}
	}
	for(int i=1;i<=q;i++){
		int x,y;
		scanf("%d %d",&x,&y);
		if(p[x]!=p[y]){
			printf("%d\n",-1);
			continue;
		}
		int stable=lca(x,y);
		printf("%d\n",stable);
	}
	return 0; 
}
2024/4/3 18:22
加载中...