题意:给一个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]);
}
}
}