CCF认证 2014-03 无线网络

这是一道最短路的变形题，在一个平面图上，两个点之间只要距离不大于r，就可以连通

所以一开始就先计算所有点（包括需要新增的点）之间的距离，如果不大于r，就添加无向边连接这两个点

在最短路搜索时，不仅要记录路劲长度，还要记录路过的新增点的数量（不能大于K），用迪杰斯特拉算法

为了保证答案正确，不陷入局部最优解，需要将一般的dist[N]，改变为dist[N][M]（N为点，M为路过的新增点个数）

最后在dist[1][0]~dist[1][K]中取最小值作为答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int N=200+10; //路由器总数量 
const int M=100+10; 
int first[N];//初始化为-1
int dist[N][M]; //最短距离，前一个代表是哪个点，后一个代表经过了多少个新增路由器 
struct Edge
{
    int to,v,next;
}edge1[2*N*N]; //邻接表
int tt; //初始化为0
void add(int a,int b) //添加边
{
    edge1[tt].to=b;
    edge1[tt].next=first[a];
    first[a]=tt++;
}
struct node
{
	int p,d,k; //位置，距离，经过的新增路由器数量 
	bool operator < (const node &u) const
	{
		return d>u.d;
	}
	node(int a,int b,int c)
	{
		p=a;d=b;k=c;
	}
};
int n,m,k;
double r,x[N],y[N]; //坐标 
int main()
{
	scanf("%d%d%d%lf",&n,&m,&k,&r);
	int cnt=0;
	memset(first,-1,sizeof(first));
	for(int i=0;i<n+m;i++)
	{
		scanf("%lf%lf",&x[cnt],&y[cnt]);
		cnt++;
		for(int j=0;j<cnt-1;j++) //检查两两之间能否连通，可以的话直接加边 
		{
			double d=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
			if(d<=r)
			{
				add(i,j);
				add(j,i);
			}
		}
	}
	priority_queue<node> q; //迪杰斯特拉 
	memset(dist,999999,sizeof(dist));
	q.push(node(0,0,0));
	memset(dist[0],0,sizeof(dist[0]));
	while(!q.empty())
	{
		node u=q.top();
		q.pop();
		for(int i=first[u.p];i!=-1;i=edge1[i].next)
		{
			int v=edge1[i].to;
			int vd=u.d+1,vk=u.k;
			if(v>=n) vk++; //这个点是新增路由器，k+1 
			if(vk>k) continue; //不能再新增了 
			if(dist[v][vk]>vd) //松弛 
			{
				dist[v][vk]=vd;
				q.push(node(v,vd,vk));
			}
		}
	}
	int ans=999999;
	for(int i=0;i<M;i++) //找到最短的路 
	ans=min(ans,dist[1][i]);
	printf("%d\n",ans-1);
    return 0;
}