poj 1511 Invitation Cards(spfa)

给一个有向图。求 1到所有点的最小距离之和与所有点到1最小距离之和相加的最小值。

用一个正向的原图做一次SPFA..再将所有边反过来做一次SPFA。

这题WA出翔了。最后发现dis数组和ans要用long long 来存。并且INF的选取也要小心，不能太大也不能太小。还有就是define INF的时候，后面要有ll，表示这是长整型。不然codeblocks里编译会有warning。但是提交不影响结果。

// spfa是BellmanFord的改进。但spfa没有判断负权回路的功能(当然可以通过记录每个点入队的次数来判断，若一个点入队超过n(结点数)次，说明有负权回路)，需要事先判断好，否则spfa不能正常工作。
//因为spfa要用到，与某个点相邻的所有边，因此建图的时候要注意保存这一信息。
//池子法里的first数组可以记录这一信息
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#define MAXNODE 1000500
#define MAXEDGE 1600000
#define INF 250000000000ll //这里要注意，不然会有warning(codeblocks)
using namespace std;
struct Edge
{
int from,to,cost,next;
} edge[MAXEDGE];
int first[MAXNODE];

queue <int> Q;
long long dis[MAXNODE];
bool inqueue[MAXNODE];
long long spfa(int n,int m)//n为点数，m为边数
{
long long ans=0;
dis[1]=0; //1是源点
memset(inqueue,0,sizeof(inqueue));
for(int i=2; i<=n; i++)
{
dis[i]=INF;
}
inqueue[1]=true;
while(!Q.empty())Q.pop();
Q.push(1);
int t;
while(!Q.empty())
{
t=Q.front();
Q.pop();
inqueue[t]=false;
if(first[t]==-1)continue;
Edge et=edge[first[t]];
while(1)
{
if(dis[et.to]>dis[et.from]+et.cost)
{
dis[et.to]=dis[et.from]+et.cost;
if(!inqueue[et.to])
{
Q.push(et.to);
inqueue[et.to]=1;
}
}
if(et.next==-1)break;
else et=edge[et.next];
}
}
for(int i=2;i<=n;i++)
{
ans+=dis[i];
}
return ans;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int n,m;
long long res=0;
scanf("%d %d",&n,&m);
memset(first,-1,sizeof(first));
for(int i=0; i<m; i++)
{
scanf("%d %d %d",&edge[i].from,&edge[i].to,&edge[i].cost); //这里存的是有向图
edge[i].next=first[edge[i].from]; //头插
first[edge[i].from]=i;//存储边的下标
}
res+=spfa(n,m);

memset(first,-1,sizeof(first));
for(int i=0;i<m;i++) //将原有向图反过来
{
int temp;
temp=edge[i].from;
edge[i].from=edge[i].to;
edge[i].to=temp;

edge[i].next=first[edge[i].from]; //头插
first[edge[i].from]=i;//存储边的下标
}
res+=spfa(n,m);
printf("%I64d\n",res);
}
return 0;
}