CCF_201403-4——无线网络（spfa）

问题描述

　　目前在一个很大的平面房间里有 n 个无线路由器,每个无线路由器都固定在某个点上。任何两个无线路由器只要距离不超过 r 就能互相建立网络连接。
　　除此以外,另有 m 个可以摆放无线路由器的位置。你可以在这些位置中选择至多 k 个增设新的路由器。
　　你的目标是使得第 1 个路由器和第 2 个路由器之间的网络连接经过尽量少的中转路由器。请问在最优方案下中转路由器的最少个数是多少?

输入格式

　　第一行包含四个正整数 n,m,k,r。(2 ≤ n ≤ 100,1 ≤ k ≤ m ≤ 100, 1 ≤ r ≤ 108)。
　　接下来 n 行,每行包含两个整数 xi 和 yi,表示一个已经放置好的无线 路由器在 (xi, yi) 点处。输入数据保证第 1 和第 2 个路由器在仅有这 n 个路由器的情况下已经可以互相连接(经过一系列的中转路由器)。
　　接下来 m 行,每行包含两个整数 xi 和 yi,表示 (xi, yi) 点处可以增设 一个路由器。
　　输入中所有的坐标的绝对值不超过 108,保证输入中的坐标各不相同。

输出格式

　　输出只有一个数,即在指定的位置中增设 k 个路由器后,从第 1 个路 由器到第 2 个路由器最少经过的中转路由器的个数。

样例输入

5 3 1 3
0 0
5 5
0 3
0 5
3 5
3 3
4 4
3 0

样例输出

2

天呐，这个代码里的long long一定不能少，上一次提交就是因为算距离的时候没有加long long 得了80分，，，

#include<iostream>
#include<stdio.h>
#include<cmath>
#include<queue>
#define N 220
#define INF 0x3f3f3f3f
using namespace std;
int n,m,k;
long long r;
queue<int> que;
int dis[N][N];//表示i点和j点之间的距离，若连通用1表示
int vis[N]={0};//点是否在队列中
int num[N]={0};//表示这个点连通到第一个点所要增加的choose的个数
int ans[N];//表示这个点到第一个点的距离

struct node
{
	int x,y;
	int state;
}a[N];

void spfa()
{
	vis[1]=1;
	que.push(1);
	ans[1]=0;
	while(!que.empty())
	{
		int x=que.front();
		que.pop();
		vis[x]=0;
		for(int i=1;i<=n+m;i++)
		{
			if(dis[x][i]!=1000&&ans[i]>ans[x]+dis[x][i])
			{
				if(a[x].state==2)
				{
				  if(num[x]+1<=k)
					num[i]=num[x]+1;
				  else
					continue;
				}
				else
				num[i]=num[x];
				ans[i]=ans[x]+dis[x][i];
				if(vis[i]==0)
				{
					vis[i]=1;
					que.push(i);
				}
			}
			else if(dis[x][i]!=1000&&ans[i]==ans[x]+dis[x][i]&&num[i]>num[x])
			{
				if(a[i].state==1)
				 num[i]=num[x];
				else
				num[i]=num[x]+1;
				if(vis[i]==0)
				{
					vis[i]=1;
					que.push(i);
				}
			}
		}
	}
}

int main()
{
   for(int i=0;i<N;i++)
   {
   	ans[i]=INF;
   	for(int j=0;j<N;j++)
   	dis[i][j]=INF;//不可连通
   }
   
   cin>>n>>m>>k>>r;
   for(int i=1;i<=n;i++)
   {
   	 cin>>a[i].x>>a[i].y;
   	 a[i].state=1;//表示已放置的
   }
   for(int i=n+1;i<=n+m;i++)
   {
   	cin>>a[i].x>>a[i].y;
   	a[i].state=2;//表示待放置的点
   }
   for(int i=1;i<=n+m;i++)
   {
   	for(int j=1;j<=n+m;j++)
   	{
   		if(j==i)
   		dis[i][j]=0;
   		else
   		{
			long long distance=(long long)(a[i].x-a[j].x)*(long long)(a[i].x-a[j].x)+(long long)(a[i].y-a[j].y)*(long long)(a[i].y-a[j].y);
			if(distance<=(long long)r*r)
			{
			  dis[i][j]=1;//表示可连通
			  dis[j][i]=1;
			}
		}
		//cout<<i<<" "<<j<<" "<<dis[i][j]<<endl;
	}
   }
   spfa();
   cout<<ans[2]-1<<endl;
   
}