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 = 10000+5;

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

struct node
{
    int v,w,next;
}e[10*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=0;i<=n;i++)  d[i] = INF;
    d[st] = 0;
    memset(done,0,sizeof(done));
    Q.push(HeapNode(st,0));
    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!=-1;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 = 0;
        memset(head,-1,sizeof(head));
        memset(flag,0,sizeof(flag));
        for(int i=0;i<=n;i++)  ans[i].reset();
        for(int i=1;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=1;i<=s;i++)
        {
            scanf("%d",&a[i]);
            addEdge(0,a[i],0);
            flag[a[i]] = true;
            ans[a[i]].set(i);  //特殊点的答案就是自己
        }
        dijkstra(0);
        while(q--)
        {
            int tot = 0;
            int x;scanf("%d",&x);
            for(int i=1;i<=s;i++)
                if(ans[x].test(i))  tmp[tot++]=a[i];
            sort(tmp,tmp+tot);
            for(int i=0;i<tot;i++)
                printf("%d%c",tmp[i],i==tot-1?'
':' ');
        }
    }
    return 0;
}