Hdu 1535 Invitation Cards

 

大意：给你一个源点，让你从这里派发n个学生去其余的n-1个站点去邀请人们去CSS，然后再返回CSS，使得总的cost最小。

 

思路：

（1）过去的时候：求一次最短路，将所有的d[i]相加。

（2）回来的时候：我开始想把SPFA所有的点，然后相加，估计会超时。由于是有向边，可以用到一个巧妙的转移的方法，我们将有向边反向，由于题目保证所有的点均可到达，所以SPFA源点，然后相加可得结果。

 

CODE：

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;

const int SIZE = 1000001;
const int INF = 0x3f3f3f3f;
int u[2*SIZE], v[2*SIZE], w[2*SIZE], next[2*SIZE];
int s_u[2*SIZE], s_v[2*SIZE], s_w[2*SIZE];
int first[SIZE], d[SIZE];
int n, m, cnt;

void init()
{
    memset(u, 0, sizeof(u));
    memset(v, 0, sizeof(v));
    memset(w, INF, sizeof(w));
    memset(d, 0, sizeof(d));
    memset(next, 0, sizeof(next));
    memset(first, -1, sizeof(first));
    cnt = 0;
}

void spfa(int src)
{
    queue<int> q;
    bool inq[SIZE] = {0};
    for(int i = 1; i <= n; i++) d[i] = (i == src)? 0:INF;
    q.push(src);
    while(!q.empty())
    {
        int x = q.front(); q.pop();
        inq[x] = 0;
        for(int e = first[x]; e!=-1; e = next[e]) if(d[v[e]] > d[x]+w[e])
        {
            d[v[e]] = d[x] + w[e];
            if(!inq[v[e]])
            {
                inq[v[e]] = 1;
                q.push(v[e]);
            }
        }
    }
}

void read_graph(int u1, int v1, int w1)
{
    u[cnt] = u1; v[cnt] = v1; w[cnt] = w1;
    next[cnt] = first[u[cnt]];
    first[u[cnt]] = cnt;
    cnt++;
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        int ans = 0;
        init();
        scanf("%d%d", &n, &m);
        for(int i = 1; i <= m; i++)    scanf("%d%d%d", &s_u[i], &s_v[i], &s_w[i]);
        for(int i = 1; i <= m; i++) read_graph(s_u[i], s_v[i], s_w[i]);
        spfa(1);
        for(int i = 1; i <= n; i++) ans += d[i];
        init();
        for(int i = 1; i <= m; i++) read_graph(s_v[i], s_u[i], s_w[i]);
        spfa(1);
        for(int i = 1; i <= n; i++) ans += d[i];
        printf("%d\n", ans);
    }
    return 0;
}