2017年湖南省赛I题（最短路）

Nearest Maintenance Point

题意：给出一个图，然后给出一些关键点，然后有很多个询问，每次询问的时候问到达某点x最近的关键点是哪一个，如果有多个关键点同样近，那么把这些点全部输出。
思路：用dijkstra算法把每个点到达关键点的最短路求出，并且用bitset类型把每个点最近的关键点标记

#include<iostream>
#include<algorithm>
#include<set>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
#include<queue>
#include<map>
#include<cmath>
#include<bitset>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int N=50005;
typedef pair<int,int> P;
struct Edge{
    int to,w;
};
int n,m,s,q,a[N];
vector<Edge> G[N];
bitset<1010> p[N];//最多1000个关键点,关键点的序号对应每一位

void add_edge(int x,int y,int z){
    Edge e1{y,z};
    G[x].push_back(e1);
    Edge e2{x,z};
    G[y].push_back(e2);
}

void dijkstra(){
    int dis[N];
    priority_queue<P,vector<P>,greater<P> > que;
    fill(dis,dis+n+1,inf);

    for(int i=1;i<=s;i++){
        p[a[i]][i]=1; //点a[i]最近的关键点是第i个关键点,所以要标记为1
        dis[a[i]]=0; 
        que.push(P(0,a[i]));
    }

    while(!que.empty()){
        int d=que.top().first,v=que.top().second;
        que.pop();
        if(d>dis[v]) continue;

        for(int i=0;i<G[v].size();i++){
            Edge e=G[v][i];
            if(dis[v]+e.w<dis[e.to]){
                dis[e.to]=dis[v]+e.w;
                que.push(P(dis[e.to],e.to));
                p[e.to]=p[v]; //最短路更新,新的状态覆盖原来的状态
            }
            else if(dis[v]+e.w==dis[e.to]){
                p[e.to]|=p[v]; //如果相等,则继承状态
            }
        }
    }
}

int main()
{

    scanf("%d%d%d%d",&n,&m,&s,&q);
    int x,y,z;

    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&x,&y,&z);
        add_edge(x,y,z);
    }
    for(int i=1;i<=s;i++){
        scanf("%d",&a[i]);
    }
    sort(a+1,a+1+n);//求关键点升序输出
    dijkstra();
    int qr;
    while(q--){
        int ans[1010],cnt=0;
        scanf("%d",&qr);
        for(int i=1;i<=s;i++){
            if(p[qr][i]==1)
                ans[++cnt]=a[i];
        }
        for(int i=1;i<=cnt;i++)
            printf("%d ",ans[i]);
        printf("\n");
    }
	return 0;
}