CLIQUED: 小熊与团间距离
题目描述
小熊国里有 N 座城市,编号为 1 ∼ N。城市间由双向道路连通。
编号为 1 ∼ K 的城市都是古城,那里的市民们喜欢秩序与规律。这些城市两两由长度为 X 的
双向道路连接,共有 K(K − 1)/2 条这样的道路。
此外,还有 M 条新建的道路,长度不一定相同。其中第 i 条链接了编号为 ai 和 bi 的城市,
长度为 ci。
不存在由一个城市连向自己的道路。所有道路(共 M + K(K − 1)/2 条)两两不同。保证可
以从一个城市出发,经由城市间的双向道路,到达任意一个其它城市。
小熊 Limak 住在编号为 S 的城市中。它想要知道,从 S 出发,到达其它每个城市的最短路径
长度是多少。你能帮他求出来吗?
输入格式
输入的第一行包含一个整数 T,代表测试数据的组数。接下来是 T 组数据。
每组数据的第一行包含五个整数 N、K、X、M 和 S,分别代表城市数、古城数、古城间的
道路长度、新建道路的条数,以及 Limak 居住在的城市编号。
接下来 M 行,每行包含三个整数 ai、bi 和 ci,代表有一条连接编号为 ai 和 bi 的城市的长度
为 ci 的新建道路。
保证不存在由一个城市连向自己的道路,所有道路两两不同,且任意城市间两两可达。
输出格式
对于每组数据,输出一行,包含 N 个整数。第 i 个整数代表从编号为 S 的城市到编号为 i 的
城市的最短路径长度。从编号为 S 的城市到自身的距离为 0。
数据范围与子任务
• 1 ≤ T ≤ 3
• 2 ≤ K ≤ N ≤ 105
• 0 ≤ M ≤ 105
• 1 ≤ S, ai
, bi ≤ N
• 1 ≤ X, ci ≤ 109
子任务 1(45 分):
• 1 ≤ K ≤ 500
子任务 2(55 分):
• 无附加限制
样例数据
输入
3
5 4 100 2 3
1 5 50
5 3 160
5 4 100 2 3
1
April Challenge 2017
1 5 50
5 3 140
8 3 15 7 5
3 4 10
4 5 20
5 6 10
6 1 7
3 7 1000
7 8 50
3 5 1000000000
输出
100 100 0 100 150
100 100 0 100 140
17 32 30 20 0 10 1030 1080
样例解释
在第一组数据中,共有 N = 5 座城市,其中前 K = 4 座城市间由长度为 X = 100 的道路两
两相连,此外还有 M = 2 条新建道路。剩下的自己画个图就会明白啦。注意输出中的第三个数为
0,即从编号为 S = 3 的城市到自己的距离为 0。
时间限制
1 秒
(思路:因为有一个完全图,两点之间一一建边肯定会T。我们可以建一个虚点0,使编号为1到k的点到0的边长为0,而0到编号为1到k的点的边长为x。以点0为媒介,来更新完全图中的点的最短距离。其实,完全图中的前k个点,只会用到第一个出现在队列中的点来更新其他的点,之后便用不到完全图中的边了。)
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>
#include <stack>
#include <set>
#include <cmath>
#include <string>
#define LL long long
using namespace std;
const int maxn=2e5+10;
const int maxm=5e5+10;
const LL inf=1e18;
int n,m,k,x,s;
int head[maxn],vis[maxn],cnt;
LL dis[maxn];
struct node
{
int to,next;
LL w;
}E[maxm];
struct Node
{
int u;
LL w;
Node(){}
Node(int u,LL w):u(u),w(w){}
friend bool operator <(Node a,Node b)
{
return a.w > b.w;
}
};
void add(int u,int v,LL w)
{
E[cnt].to=v;
E[cnt].w=w;
E[cnt].next=head[u];
head[u]=cnt++;
}
void dijs(int s)
{
priority_queue<Node> q;
int u,v;
LL nd;
dis[s]=0;
q.push(Node(s,0));
while(q.size())
{
Node now=q.top();
q.pop();
u=now.u;
nd=now.w;
if(vis[u]) continue;
vis[u]=1;
//注意vis的位置,和bfs不同!!!!
for(int i=head[u];i!=-1;i=E[i].next)
{
v=E[i].to;
if(vis[v]) continue;
if(dis[v]>nd+E[i].w)
{
dis[v]=nd+E[i].w;
q.push(Node(v,dis[v]));
}
}
}
for(int i=1;i<=n;i++)
{
printf("%lld%c",dis[i],i==n?'\n':' ');
}
}
int main(void)
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d%d%d",&n,&k,&x,&m,&s);
cnt=0;
for(int i=0;i<=n;i++)
{
head[i]=-1;
dis[i]=inf;
vis[i]=0;
}
for(int i=1;i<=k;i++)
{
add(0,i,x);
add(i,0,0);
}
int u,v;
LL w;
for(int i=0;i<m;i++)
{
scanf("%d%d%lld",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
dijs(s);
}
return 0;
}