题目:
有 N 个村庄,编号 1 到 N。
村庄之间有 M 条无向道路,第 i 条道路连接村庄 ai 和村庄 bi,长度是 ci。
所有村庄都是连通的。
共有 K 个村庄有商店,第 j 个有商店的村庄编号是 xj。
然后给出 Q 个询问,第 k 个询问给出一个村庄的编号 yk,问该村庄距离最近的商店有多远?
输入格式
第一行包含两个整数 N,M。
接下来 M 行,每行包含三个整数 ai,bi,ci,表示第 i 条道路连接村庄 ai 和村庄 bi,长度是 ci。
再一行包含整数 K。
接下来 K 行,每行包含一个整数 xj,表示第 j 个有商店的村庄编号是 xj。
再一行包含整数 Q。
接下来 Q 行,每行包含一个整数 yk,表示询问编号为 yk 的村庄与其距离最近的商店之间的距离。
输出格式
对于每个询问,输出该询问的结果。
数据范围
2≤N≤105,
N−1≤M≤min(N(N−1)2,105),
1≤Q≤105,
1≤K≤N,
1≤ci≤10000
建立虚拟源点的步骤可以省去,直接将所有的起点入队,距离设置为0。
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N = 1e5+10,M = 1e5+10;
typedef pair<int,int> PII;
int e[2*M],ne[2*M],h[N],w[2*M],idx;
int dist[N];
int id[N];
bool st[N];
int n,m;
void add(int a,int b,int c)
{
e[idx] = b,ne[idx] = h[a],w[idx] =c,h[a] = idx++;
}
int main()
{
scanf("%d%d",&n,&m);
memset(h,-1,sizeof h);
for(int i = 0; i<m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(a,b,c),add(b,a,c);
}
int k;
scanf("%d",&k);
memset(dist,0x3f,sizeof dist);
priority_queue<PII,vector<PII>,greater<PII> >heap;
for(int i = 0;i<k;i++)
{
int x;
scanf("%d",&x);
dist[x] = 0;
heap.push({0,x});
}
while(heap.size())
{
auto t= heap.top();
heap.pop();
int ver = t.second,distance = t.first;
if(st[ver]) continue;
st[ver] = true;
for(int i = h[ver];~i;i=ne[i])
{
int j =e[i];
if(dist[j]>distance+w[i])
{
dist[j] = distance+w[i];
heap.push({dist[j],j});
}
}
}
int q;
scanf("%d",&q);
while(q--)
{
int x;
scanf("%d",&x);
printf("%d\n",dist[x]);
}
return 0;
}
标签:ci,dist,idx,int,短路,源点,dijkstra,村庄,include
From: https://www.cnblogs.com/freshman666/p/17174839.html