poj 1511 Invitation Cards && poj 3268 Silver Cow Party （spfa）

两道题类似，这只给出1511的代码。

//poj 1511
/*题意：p个结点，q条单向的路径，1是起点，求1到任意点，并从该点回到1的花费和要最小。
1.点太多了，dijkstra和floyd都不行。2.spfa，注意求返程时，只需把所有路径的起点和终点颠倒。
3.数据很大，不用int，用long long 或者__int64;
*/

#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;
#define M 1000000000
int n,p,q;
__int64 a[1000001],b[1000001],c[1000001],dis[1000001];
struct node
{
    int s,e;
    __int64 w;
};
vector<node>edge[1000001];
queue<int>Q;

void spfa(int s)
{
    while(!Q.empty())
    Q.pop();

    int i;
    for(i=1;i<=p;i++)
    dis[i]=M;
    dis[s]=0;

    Q.push(s);
    while(!Q.empty())
    {
        int w=Q.front();
        Q.pop();
        for(i=0;i<edge[w].size();i++)
        {
            node r=edge[w][i];
            if(dis[r.e]>dis[w]+r.w)
            {
                dis[r.e]=dis[w]+r.w;
                Q.push(r.e);
            }
        }
    }
}

int main()
{
    scanf("%d",&n);
    while(n--)
    {
        int i;
        __int64 sum=0;
        for(i=1;i<=1000000;i++)
        edge[i].clear();

        scanf("%d%d",&p,&q);
        for(i=1;i<=q;i++)
        {
            scanf("%I64d%I64d%I64d",&a[i],&b[i],&c[i]);
            node f;
            f.s=a[i];
            f.e=b[i];
            f.w=c[i];
            edge[a[i]].push_back(f);
        }
        spfa(1);
        for(i=1;i<=p;i++)
        sum+=dis[i];

        for(i=1;i<=1000000;i++)
        edge[i].clear();
        for(i=1;i<=q;i++)
        {
            node g;
            g.s=b[i];
            g.e=a[i];
            g.w=c[i];
            edge[b[i]].push_back(g);
        }
        spfa(1);
        for(i=1;i<=p;i++)
        sum+=dis[i];

        printf("%I64d\n",sum);
    }
    return 0;
}