CSU 2005 Nearest Maintenance Point（最短路+bitset）

https://vjudge.net/problem/CSU-2005

题意：
给出带权值的图，图上有一些特殊点，现在给出q个询问，对于每个询问，输出离该点最近的特殊点，如果有多个，则按升序输出。

思路：
因为有多次查询，不可能对于每个询问都去跑一遍最短路。必须以特殊点为起点跑一遍最短路，但是这样路径的记录就是问题了。正解是用bitset来记录状态，在最短路松弛更新状态时，继承前驱节点即可。

 #include<iostream>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<queue>
 #include<bitset>
 #include<set>
 using namespace std;
 const int INF = 0x3f3f3f3f;
 const int maxn = +;

 int n,m,s,q,tot;
 int head[maxn],d[maxn],a[];
 bool done[maxn],flag[maxn];
 bitset<> ans[maxn];
 int tmp[];

 struct node
 {
     int v,w,next;
 }e[*maxn];

 void addEdge(int u, int v, int w)
 {
     e[tot].v = v;
     e[tot].w = w;
     e[tot].next = head[u];
     head[u] = tot++;
 }

 struct HeapNode
 {
     int u,d;
     HeapNode(int u, int d):u(u),d(d){}
     bool operator< (const HeapNode& rhs) const
     {
         return d > rhs.d;
     }
 };

 void dijkstra(int st)
 {
     priority_queue<HeapNode> Q;
     for(int i=;i<=n;i++)  d[i] = INF;
     d[st] = ;
     memset(done,,sizeof(done));
     Q.push(HeapNode(st,));
     while(!Q.empty())
     {
         HeapNode x = Q.top(); Q.pop();
         int u = x.u;
         if(done[u])  continue;
         done[u] = true;
         for(int i=head[u];i!=-;i=e[i].next)
         {
             int v = e[i].v;
             if(d[v] > d[u]+e[i].w)
             {
                 d[v] = d[u] + e[i].w;
                 if(!flag[v]) ans[v] = ans[u]; //如果不是特殊点，就更新
                 Q.push(HeapNode(v,d[v]));
             }
             else if(d[v] == d[u]+e[i].w)  ans[v]|=ans[u];  //相等的话，就加上当前点的答案
         }
     }
 }

 int main()
 {
     //freopen("in.txt","r",stdin);
     while(~scanf("%d%d%d%d",&n,&m,&s,&q))
     {
         tot = ;
         memset(head,-,sizeof(head));
         memset(flag,,sizeof(flag));
         for(int i=;i<=n;i++)  ans[i].reset();
         for(int i=;i<=m;i++)
         {
             int u,v,w;
             scanf("%d%d%d",&u,&v,&w);
             addEdge(u,v,w);
             addEdge(v,u,w);
         }
         for(int i=;i<=s;i++)
         {
             scanf("%d",&a[i]);
             addEdge(,a[i],);
             flag[a[i]] = true;
             ans[a[i]].set(i);  //特殊点的答案就是自己
         }
         dijkstra();
         while(q--)
         {
             int tot = ;
             int x;scanf("%d",&x);
             for(int i=;i<=s;i++)
                 if(ans[x].test(i))  tmp[tot++]=a[i];
             sort(tmp,tmp+tot);
             for(int i=;i<tot;i++)
                 printf("%d%c",tmp[i],i==tot-?'\n':' ');
         }
     }
     return ;
 }