2005: Nearest Maintenance Point

题目链接:http://acm.csu.edu.cn/csuoj/problemset/problem?pid=2005

题意：

有一个n个顶点m条边的无向图。在这n个点中有s个顶点是维修站，然后q次询问，每次询问包含一个顶点u，输出距离u最近的加油站。 

如果有多个加油站，按照节点编号从小到大输出。 

数据范围：(2 ≤ n ≤ 104, 1 ≤ m ≤ 5 × 104),(1 ≤ s ≤ minn, 1000,1 ≤ q ≤ minn, 1000)

分析: 将加油站都放入起始队列，距离设为0 ， 就可以保证 d[v] 始终是离当前点最近的加油站，在更新过程中使用bitset 不断统计 到该点距离相同的加油站有几个（用bitset保存）

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
#include <bitset>
using namespace std;
#define maxn 20500
typedef pair<int,int> PII;
struct Edge{int y,w;};
bitset<1010>bs[maxn];
vector<PII>E[maxn];
#define INF 1e9

int sx[maxn];
int n , m , s , q;
int d[maxn];

void dijkstra()
{
    for(int i = 0 ; i <= n ; ++i)d[i] = INF , bs[i].reset();
    priority_queue<PII,vector<PII>,greater<PII> >Q;
    for(int i = 1 ; i <= s ; ++i)
    {
        Q.push( PII( 0 , sx[i] ) );
        d[ sx[i] ] = 0;
        bs[ sx[i] ][i] = 1;
    }
    while( !Q.empty() )
    {
        int u = Q.top().second;
        int c = Q.top().first;
        Q.pop();
        if( c > d[u] )continue;

        for(int i = 0 ; i < E[u].size() ; ++i)
        {
            int v = E[u][i].first;
            int w = E[u][i].second;

            if( d[v] > d[u] + w )
            {
                d[v] = d[u] + w;
                bs[v] = bs[u];
                Q.push(PII(d[v],v));
            }
            else if( d[v] == d[u] + w )
            {
                bs[v] |= bs[u];
            }
        }
    }

}
int ans[maxn];
void solve()
{
    while(q--)
    {
        int x,tot=0; scanf("%d",&x);
        for(int i=1;i<=s;i++)
            if (bs[x][i]) ans[++tot]=sx[i];       //判断对应位置是否能以最短路到达
        for(int i=1;i<tot;i++)
            printf("%d ",ans[i]);
        printf("%d\n",ans[tot]);
    }
}
int main()
{
    while( cin >> n >> m >> s >> q )
    {
        for(int i = 0 ; i <= n ; ++i)E[i].clear();
        for(int i = 0 ;  i < m ; ++i)
        {
            int a,b,c;
            scanf("%d %d %d",&a,&b,&c);
            E[a].push_back(PII(b,c));
            E[b].push_back(PII(a,c));
        }

        for(int i = 1 ; i <= s ; ++i){
            scanf("%d",sx+i);
        }
        sort(sx+1,sx+s+1);
        dijkstra();
        solve();
    }
}