CSU 2005 Nearest Maintenance Point (bitset + 迪杰斯特拉）

题目链接

题意：一个省由通过一些双向道路连接的n个城市（标记为1、2，…，  n）组成。每条道路连接一对不同的城市。一家机器人公司在一些城市建立了维修点。现在，他们需要您的帮助来编写程序来查询距特定城市最近的维护点，如果最近的维修点有多个，打印出所有的维修点。

思路：用bitset保存最短路的状态，如果最短路有多个，即d[v] == dist + d[u],那么bitset或运算算出即可。

bitset的几个常用函数，reset（清0），set（清1），更多详细用法参考：https://www.cnblogs.com/magisk/p/8809922.html

简介版简介：

#include <bits/stdc++.h>
#define pi pair<int,int>
#define mk make_pair
using namespace std;
const int maxn = 10010,inf = 1e9+10;

vector<pi>G[maxn];
bitset<1010>p[maxn];
int a[maxn],ans[maxn],d[maxn],vis[maxn];

void dij(int s,int n)
{
	for(int i=1;i<=n;i++)
	{
		d[i] = inf;
		vis[i] = 0;
		p[i].reset();
	}
	priority_queue<pi>q;
	for(int i=1;i<=s;i++)
	{
		d[a[i]] = 0;
		p[a[i]][i] = 1;
		q.push(mk(0,a[i])); 
	}
	while(!q.empty())
	{
		pi tmp = q.top();
		q.pop();
		int u = tmp.second;
		if(vis[u])continue;
		vis[u] = 0;
		for(int i=0;i<G[u].size();i++)
		{
			tmp = G[u][i];
			int v = tmp.first;
			int w = tmp.second;
			if(d[v] > d[u] + w)
			{
				d[v] = d[u]+w;
				p[v] = p[u];
				q.push(mk(-d[v],v));  
			}
			else if(d[v] == d[u] + w)
			{
				p[v]|=p[u];
			}
		}
	}
}
int main()
{
	int n,m,s,q;
	while(~scanf("%d%d%d%d",&n,&m,&s,&q))
	{
		for(int i=1;i<=m;i++)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			G[u].push_back(mk(v,w));
			G[v].push_back(mk(u,w));
		}
		for(int i=1;i<=s;i++)scanf("%d",a+i);
		dij(s,n);
		while(q--)
		{
			int x,cnt = 0;
			scanf("%d",&x);
			for(int i=1;i<=s;i++)
			{
				if(p[x][i])ans[++cnt] = a[i];
			}
			for(int i=1;i<cnt;i++)printf("%d ",ans[i]);
			printf("%d\n",ans[cnt]);
		}
	}
	return 0;
}