CodeChef - CLIQUED Bear and Clique Distances(建虚点+最短路)

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;
}