hdu 1535 Invitation Cards ( 做2次spfa )

1 题目大意

给你一个源点，让你从这里派发学生到其余的点邀请人们去CSS，然后这些学生再返回源点，求使得学生总的cost最小的值。

2 解题思路

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

（2）从其他点回来的时候：将有向边反向，由于题目保证所有的点均可到达，所以SPFA源点，然后相加可得结果。

3 代码

#include<stdio.h>

#include<string.h>

#include<queue>

using namespace std;

#define MAXV 1000000

#define MAXE 1000000

#define MAXINT 999999999

typedef struct

{

    int from;

    int to;

    int weight;

    int next1;

    int next2;

}Edge;

int nv, ne, tot;

int dis[MAXV], vis[MAXV], index1[MAXV], index2[MAXV];

Edge edge[MAXE];

void init()

{

    tot=0;

    memset(index1, -1, sizeof(index1));

    memset(index2, -1, sizeof(index2));

}

void insert_edge(int s, int t, int w)

{

    edge[tot].from=s;

    edge[tot].to=t;

    edge[tot].weight=w;

    edge[tot].next1=index1[s];

    edge[tot].next2=index2[t];

    index1[s]=index2[t]=tot;

    tot++;

}

long long spfa(int s, int t)

{

    int i, u, v, w;

    long long sum;

    queue<int> que;

    sum=0;

    //第一次spfa，求源点到所有点的最短距离

    for(i=0;i<=nv;i++)

    {

        dis[i]=MAXINT;

        vis[i]=0;

    }

    dis[s]=0;

    que.push(s);

    vis[s]=1;

    while(!que.empty())

    {

        u=que.front();

        que.pop();

        vis[u]=0;

        for(i=index1[u];i!=-1;i=edge[i].next1)

        {

            v=edge[i].to;

            w=edge[i].weight;

            if(dis[v]>dis[u]+w)

            {

                dis[v]=dis[u]+w;

                if(!vis[v])

                {

                    que.push(v);

                    vis[v]=1;

                }

            }

        }

    }

    //求源点到其他点的距离之和

    for(i=1;i<=nv;i++)

    {

        if(dis[i]<MAXINT)

            sum+=dis[i];

    }

    //第二次spfa，求所有点到源点的最短距离

    for(i=0;i<=nv;i++)

    {

        dis[i]=MAXINT;

        vis[i]=0;

    }

    dis[s]=0;

    que.push(s);

    vis[s]=1;

    while(!que.empty())

    {

        u=que.front();

        que.pop();

        vis[u]=0;

        for(i=index2[u];i!=-1;i=edge[i].next2)

        {

            v=edge[i].from;

            w=edge[i].weight;

            if(dis[v]>dis[u]+w)

            {

                dis[v]=dis[u]+w;

                if(!vis[v])

                {

                    que.push(v);

                    vis[v]=1;

                }

            }

        }

    }

    //求其他点到源点的距离之和

    for(i=1;i<=nv;i++)

    {

        if(dis[i]<MAXINT)

            sum+=dis[i];

    }

    return sum;

}

int main()

{

    int T;

    scanf("%d", &T);

    while(T--)

    {

        scanf("%d%d", &nv, &ne);

        int a, b, c, i;

        init();

        for(i=0;i<ne;i++)

        {

            scanf("%d%d%d", &a, &b, &c);

            insert_edge(a, b, c);

        }

        printf("%lld\n", spfa(1, nv));

    }

    return 0;

}