poj 1511 Invitation Cards

  这个题的题意就是从1到所有点以及从所有点回到1所需最小的花费。这道题实质是求最短路径，我的思路是用SPFA，正反两次SPFA。对spfah还是第一次写，不知道怎么优化。

  代码：

#include<iostream>
#include<queue>
using namespace std;
const int maxn=1000005,inf=1000000005;
struct node
{
       int y,w,next;
}e[1000005];
int h[maxn],w[maxn],x[maxn],y[maxn],d[maxn];
int visit[maxn];
int n,m,p;
long long ans;
queue<int>q;
void init(int x,int y,int w)
{
    e[++p].y=y;
    e[p].w=w;
    e[p].next=h[x];
    h[x]=p;
}
 void spfa()
 {
      int i,x,y;
      for(i=1;i<=n;i++)
      {
         d[i]=inf;
         visit[i]=0;
      }
      q.push(1);
      visit[1]=1;
      d[1]=0;
      while(!q.empty())
      {
         x=q.front();
         q.pop();
         for(i=h[x];i;i=e[i].next)
         {
            y=e[i].y;
            if(e[i].w+d[x]<d[y])
            {
              d[y]=e[i].w+d[x];
              if(!visit[y])
              {
                visit[y]=1;
                q.push(y);
              }
            }
         }
      }
 }
          
int main()
{
    int t,i,j;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%d%d",&n,&m);
       p=0;
       memset(h,0,sizeof(h));
       for(i=0;i<m;i++)
       {
         scanf("%d%d%d",&x[i],&y[i],&w[i]);
         init(x[i],y[i],w[i]);
       }
       spfa();
       ans=0;
       for(i=1;i<=n;i++)
         ans+=d[i];
       memset(h,0,sizeof(h));
       p=0;
       for(i=0;i<m;i++)
         init(y[i],x[i],w[i]);
       spfa();
       for(i=1;i<=n;i++)
         ans+=d[i];
       printf("%I64d\n",ans);  
    } 
    //system("pause"); 
    return 0;  
}