80分求助
查看原帖
80分求助
255077
麦克斯韦の妖楼主2021/9/22 23:28
#include<iostream>
#include<cstdio>
#include<cmath>
#include<set>
#include<string.h>
#include<queue>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int N=1000+10;
const int INF=0x3f3f3f3f;
int s,p;
int dx[N],dy[N];
struct Node{
	int to;
	int next;
	double w; 
}edge[N*N];
int head[N],f[N];
int tot;
void init(){
    memset(head,-1,sizeof(head));
    for(int i=0;i<=N;i++)
    {
    	f[i]=i;
	} 
	tot=0;
}
void add(int u,int v,double w)
{
	edge[tot].to=v;
	edge[tot].w=w;
	edge[tot].next=head[u];
	head[u]=tot++;
}
bool cmp(Node a,Node b)
{
	return a.w<b.w;
}
double dist(int x,int y,int xx,int yy)
{
	return sqrt((x-xx)*(x-xx)+(y-yy)*(y-yy));
}
int Find(int x)
{
	if(x==f[x])return x;
	return f[x]=Find(f[x]);
}
void Union(int u,int v)
{
	if(Find(u)!=Find(v))
	{
		f[Find(u)]=Find(v);
	}
}
inline double Max(double a,double b)
{
	return a>b?a:b;
}
double Kruskal()
{
	double ans=0.00;
	sort(edge,edge+tot,cmp);
	int tmp=0;
	for(int i=0;i<tot;i++)
	{
		int u=edge[i^1].to;
		int v=edge[i].to;
		if(Find(u)!=Find(v))
		{
			tmp++;
			ans=edge[i].w;
			Union(u,v);
			if(tmp>=p-s)
			{
				break;
			}
		}
	}
	return ans;
}
int main()
{
	init();
	scanf("%d%d",&s,&p);
	for(int i=1;i<=p;i++)
	{
		int x,y; 
		scanf("%d%d",&x,&y);
		dx[i]=x;
		dy[i]=y;
	}
	if(s==1)s=0;
	for(int i=1;i<=p;i++)
	for(int j=i+1;j<=p;j++)
	{
		double d=dist(dx[i],dy[i],dx[j],dy[j]);
		add(i,j,d);
		add(j,i,d);
	}
	printf("%.2f\n",Kruskal());
}

如上,最后一个点WA了。 然而,把

if(tmp>=p-s)
{
	break;
}

改为

if(tmp>=p-s+1)
{
	break;
}

可以过最后一个点和第3个点,其他WA了

2021/9/22 23:28
加载中...