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

CSU 2005

题意：给一个n个节点m条边的带权无向图，给出s个援助节点，有q个问题，每次问一个节点周边最短距离的援助节点有哪些。

思路：去年省赛的题，当时不幸拿了铁....看了csuzhucong这位大佬的代码才学会的，是个很好的题，把s个援助节点当起点，搜到所有节点的最短路，用bitset的位运算传递状态，如果有多个起点的某个节点的最短距离相同，那么这个节点周围最近的援助点就是那多个起点，并且用或运算保存，如果只有一个起点到某个节点的距离最短，那么该节点最近的援助节点就是那个节点，直接赋值传递状态。

#include<cstdio>
#include<bitset>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=1010;
const int inf=1e8;
struct node
{
	int u,len;
	bool friend operator<(node c,node d)
	{
		return c.len>d.len;
	}
}no,ne;
vector<int>G[maxn*20],dis[maxn*20];
bitset<1010>p[maxn*20];
int n,s,d[maxn*20],a[maxn],vis[maxn*20],ans[maxn];
void init()
{
	for(int i=0;i<=n;i++)
	{
		G[i].clear();
		dis[i].clear(); 
	}
}
void dij()
{
	priority_queue<node>q;
	for(int i=1;i<=n;i++)
	{
		d[i]=inf;
		vis[i]=0;
		p[i].reset();
	}
	for(int i=1;i<=s;i++)
	{
		no.u=a[i];
		no.len=d[a[i]]=0;
		p[a[i]][i]=1;
		q.push(no);
	}
	while(!q.empty())
	{
		no=q.top();q.pop();
		if(vis[no.u])
		continue;
		vis[no.u]=1;
		int u=no.u;
		for(int i=0;i<G[u].size();i++)
		{
			int v=G[u][i];
			int dist=d[u]+dis[u][i];
			if(d[v]>dist)
			{
				d[v]=ne.len=dist;
				ne.u=v;
				q.push(ne);
				p[v]=p[u];
			}
			else if(d[v]==dist)
			p[v]|=p[u];
		}
	}
}
int main()
{
	int m,q,u,v,len;
	while(~scanf("%d%d%d%d",&n,&m,&s,&q))
	{
		init();
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&u,&v,&len);
			G[u].push_back(v);
			G[v].push_back(u);
			dis[u].push_back(len);
			dis[v].push_back(len);
		}
		for(int i=1;i<=s;i++)
		scanf("%d",&a[i]);
		sort(a+1,a+1+s);
		dij();
		while(q--)
		{
			int tot=0;
			scanf("%d",&u);
			for(int i=1;i<=s;i++)
			if(p[u][i])
			ans[++tot]=a[i];
			for(int i=1;i<tot;i++)
			printf("%d ",ans[i]);
			printf("%d\n",ans[tot]);
		}
	}
}