【最短路 && 把一些点看成一个团】CodeChef - CLIQUED Bear and Clique Distances

Problem Desription

输入的第一行包含一个整数T，代表测试数据的组数。接下来是T组数据。
每组数据的第一行包含五个整数N、K、X、M和S，分别代表城市数、古城数、古城间的道路长度、新建道路的条数，以及Limak居住在的城市编号。接下来M行，每行包含三个整数ai、bi和ci，代表有一条连接编号为ai和bi的城市的长度为ci的新建道路。保证不存在由一个城市连向自己的道路，所有道路两两不同，且任意城市间两两可达。编号为1-K的城市都是古城，那里的市民们喜欢秩序与规律。这些城市两两由长度为X的双向道路连接，共有K(K - 1) / 2条这样的道路

Input

3
5 4 100 2 3
1 5 50
5 3 160
5 4 100 2 3
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

Output

100 100 0 100 150
100 100 0 100 140
17 32 30 20 0 10 1030 1080

思路：把古城看做一个团，因为古城里面城市之间的距离是互相相等的，设一个点0，使得0到古城任何任何一个城市，任何一个城市到0的点的距离和为X。这样就可以大大减少建边的条数

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f3f3f
using namespace std;
struct node
{
    long long to, w, next;
};
node e[1100012]; long long head[1110012];
long long cnt, n, k, x, m, s;
long long dist[100012], vis[100012];
void add(long long u, long long v, long long w)//前向星
{
    e[cnt].to = v; e[cnt].w = w; e[cnt].next = head[u];
    head[u] = cnt++;
}
void spfa()
{
    memset(vis, 0, sizeof(vis));
    memset(dist, INF, sizeof(dist));
    dist[s] = 0; vis[s] = 1; queue<int> q;
    q.push(s);
    while(!q.empty())
    {
        int u = q.front(); q.pop();
        vis[u] = 0;
        for(int i = head[u]; ~i; i = e[i].next)
        {
            int x = e[i].to, y = e[i].w;
            if(dist[x] > dist[u] + y)
            {
                dist[x] = dist[u] + y;
                if(!vis[x])
                {
                    vis[x] = 1;
                    q.push(x);
                }
            }
        }
    }
    for(int i = 1; i <= n; i++)
    {
        printf("%lld", dist[i]);
        if(i == n) printf("\n");
        else printf(" ");
    }
}
int main()
{
    int T;
    long long u, v, w;
    scanf("%d", &T);
    while(~scanf("%lld %lld %lld %lld %lld", &n, &k, &x, &m, &s))
    {
        memset(head, -1, sizeof(head));
        cnt = 0;
        while(m--)
        {
            scanf("%lld %lld %lld", &u, &v, &w);
            add(u, v, w); add(v, u, w);
        }
        for(int i = 1; i <= k; i++)//核心，0到古城 + 古城到0的距离为x
        {
            add(0, i, x / 2);
            add(i, 0, x / 2 + x % 2);
        }
        spfa();//然后就是裸的spfa
    }
    return 0;
}