poj1511 spfa

 

 

  如题：http://poj.org/problem?id=1511

用spfa求一个图1到各个点的最短路径，然后将图反向，再求一次，2次的结果相加。

 

#include<iostream>
using namespace std;

#define inf 1000000000000LL
#define MAXN 1000005
#define MAXM 1000005
struct note
{
 int u,v;
 long long c;
 note(){}
 note(int u,int v,long long c):u(u),v(v),c(c){}
};

note p[MAXM];
int head[MAXN],next[MAXM];
long long dp[MAXN];
bool map[MAXN];
int now[MAXN];
int e,top,n,m;
void addnote(int u,int v,long long c)
{
 p[e]=note(u,v,c);
 next[e]=head[u];
 head[u]=e++;
}
void init()
{
 memset(head,-1,sizeof(head[0])*(n+2));
 memset(next,-1,sizeof(next[0])*(m+2));
 e=0;
 int i;
 for(i=0;i<m;i++)
 {
  int u,v;
  long long c;
  scanf("%d%d%lld",&u,&v,&c);
  addnote(u,v,c);
 }
}
void uninit()
{
 memset(head,-1,sizeof(head[0])*(n+2));
 memset(next,-1,sizeof(next[0])*(m+2));
 e=0;
 int i;
 for(i=0;i<m;i++)
  addnote(p[i].v,p[i].u,p[i].c);
}
long long get_sum(long long a[])
{
 long long res=0;
 for(int i=1;i<=n;i++)
 {
  res+=a[i];
 }
 return res;
}
bool relax(int u,int v,long long c)
{
 if(dp[v]>dp[u]+c)
 {
  dp[v]=dp[u]+c;
  return true;
 }
 return false;
}

void spfa(int t)
{
 memset(map,0,sizeof(map[0]*(n+2)));
 int i;
 for(i=1;i<=n;i++)
  dp[i]=inf;
 dp[t]=0;
 map[t]=true;
 top=0;
 now[top++]=t;
 while(top)
 {
  int pre=now[--top];
  map[pre]=false;
  int j;
  for(j=head[pre];j+1;j=next[j])
  {
   if(relax(p[j].u,p[j].v,p[j].c)&&map[p[j].v]==false)
   {
    now[top++]=p[j].v;
    map[p[j].v]=true;
   }
  }
 }
}

int main()
{
  //freopen("C:\\1.txt","r",stdin);
 int T;
 scanf("%d",&T);
 while(T--)
 {
  scanf("%d%d",&n,&m);
  init();
  spfa(1);
  long long ans=get_sum(dp);
  uninit();
  spfa(1);
  ans+=get_sum(dp);
  printf("%lld\n",ans);
 }
}