民间数据WA了一个,求助!
查看原帖
民间数据WA了一个,求助!
496615
huajx楼主2023/10/9 22:25

记录https://www.luogu.com.cn/record/128455370 求调

#include<bits/stdc++.h>
using namespace std;
long long read(){long long x=0,sgn=1;char ch=getchar();while(ch<'0' || ch>'9'){if(ch=='-')sgn=-1;ch=getchar();}while(ch>='0' && ch<='9'){x=(x<<3)+(x<<1)+(ch&15);ch=getchar();}return x*sgn;}
void write(long long n,bool p){if(n<0){putchar('-');n=-n;}if(n==0){if(p)putchar('0');return;}write(n/10,0);putchar(n%10+'0');}
long long n,m,k,v[20005],nex[20005],poi[2505],f[2505][2505],num[2505],x,y,p,sum[2505][3],MAX=1,w[2505][3],ans;
long long aa,bb,cc,dd;
bool e[2505][2505];
queue<long long> q;
void bfs(long long z){
	q.push(z);
	while(!q.empty()){
		x=q.front();
		q.pop();
		for(int i=poi[x];i;i=nex[i]){
			if(f[z][v[i]]==MAX){
				f[z][v[i]]=f[z][x]+1;
				e[z][v[i]]=1;
				if(f[z][v[i]]<k)q.push(v[i]);
			}
		}
	}
	return;
}
int main(){
	//freopen("holiday3.in","r",stdin);
	n=read();m=read();k=read()+1;
	MAX<<=62;
	MAX=(MAX-1<<1)+1;
	for(int i=2;i<=n;i++)num[i]=read();
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++){
			f[i][j]=MAX;
		}
	for(int i=1;i<=n;i++){
		f[i][i]=0;
	}
	for(int i=1;i<=m;i++){
		x=read();y=read();
		v[++p]=y;nex[p]=poi[x];poi[x]=p;
		v[++p]=x;nex[p]=poi[y];poi[y]=p;
	}
	for(int i=1;i<=n;i++)bfs(i);
	for(int j=2;j<=n;j++){
		for(int i=2;i<=n;i++){
			if(i==j)continue;
			if(e[1][i]&&e[i][j]){
				x=num[i]+num[j];
				if(x>sum[j][0]){
					sum[j][2]=sum[j][1];
					//sum[j][2]=num[j]+num[w[j][1]];
					w[j][2]=w[j][1];
					sum[j][1]=sum[j][0];
					//sum[j][1]=num[j]+num[w[j][0]];
					w[j][1]=w[j][0];
					sum[j][0]=x;
					w[j][0]=i;
				}else if(x>sum[j][1]){
					sum[j][2]=sum[j][1];
					//sum[j][2]=num[j]+num[w[j][1]];
					w[j][2]=w[j][1];
					sum[j][1]=x;
					w[j][1]=i;
				}else if(x>sum[j][2]){
					sum[j][2]=x;
					w[j][2]=i;
				}
			}
		}
	}
	for(int i=2;i<=n;i++){
		for(int j=2;j<=n;j++){
			if(e[i][j]){
				for(int xx=0;xx<3;xx++){
					for(int yy=0;yy<3;yy++){
						if(i!=w[j][yy]&&j!=w[i][xx]&&w[i][xx]!=w[j][yy]){
							//if(sum[i][xx]+sum[j][yy]>ans){aa=w[i][xx];bb=i;cc=j;dd=w[j][yy];}
							ans=max(ans,sum[i][xx]+sum[j][yy]);
							//ans=max(ans,num[w[i][xx]]+num[i]+num[j]+num[w[j][yy]]);
						}
					}
				}
			}
		}
	}
	printf("%lld",ans);
	//printf("\n%lld %lld %lld %lld",aa,bb,cc,dd);
	return 0;
}
2023/10/9 22:25
加载中...