【ccf-csp题解】第1次csp认证-第四题-无线网络-特殊点个数限制的单源最短路径

题目描述

思路讲解

可以把题目抽象为：
从第1个点到第2个点，经过特殊点的数量不超过k的单源最短路径（其中每条边的权重均为1）

可以使用bfs解决这个问题，但是dist[][]数组和队列中放置的pair<int,int>元素不再是单单的x和y，而是点的标号b和状态d

所谓点的标号，可以事先创建一个PII point[N],这样一个下标就代表一个点

状态的意思是：从1到点b总共经过了几个特殊点？

举个栗子：

dist[5][3]：从1到5经过了3个特殊点的最短路径

由于r的限制，不是每一个都可以到达另一个点，那么这里就涉及一个技巧，即提前判断可以互相到达的点，然后把它们相连，最后就构造出一个互通且边权重为1的图：

for(int i=1;i<=n+m;i++)
	for(int j=1;j<=n+m;j++)
		if(check(i,j,r))//判断点i和点j距离是否小于等于r 
		add(i,j),add(j,i);

满分代码

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int N=210,M=N*N;
int h[N],e[M],ne[M],idx;
typedef long long LL;
typedef pair<int,int>PII;
PII p[N];
int dist[N][N];
int n,m,k,r;
bool check(int i,int j,int r)
{
	int dx=p[i].first-p[j].first;
	int dy=p[i].second-p[j].second;
	return (LL)dx*dx+(LL)dy*dy<=(LL)r*r;
}
void add(int a,int b)
{
	e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
int bfs()
{
	memset(dist,0x3f,sizeof dist);
	queue<PII>q;
	q.push({1,0});
	dist[1][0]=0;
	while(q.size())
	{
		auto t=q.front();
		q.pop();
		for(int i=h[t.first];i!=-1;i=ne[i])
		{
			int a=e[i],b=t.second;
			if(a>n)b++;
			if(b<=k)
			{
				if(dist[a][b]>dist[t.first][t.second]+1)
				{
					dist[a][b]=dist[t.first][t.second]+1;
					q.push({a,b});
				}
			}
		}
	}
	int res=1e8;
	for(int i=0;i<=k;i++)
	res=min(res,dist[2][i]);
	return res-1;
}
int main()
{
	cin>>n>>m>>k>>r;
	memset(h,-1,sizeof h);
	for(int i=1;i<=n;i++)cin>>p[i].first>>p[i].second;
	for(int i=n+1;i<=n+m;i++)cin>>p[i].first>>p[i].second;
	for(int i=1;i<=n+m;i++)
		for(int j=1;j<=n+m;j++)
			if(check(i,j,r))//判断点i和点j距离是否小于等于r 
			add(i,j),add(j,i);
	cout<<bfs()<<endl;
	return 0;
}