pku1511_spfa*2+queue

pku1511_spfa*2+queue

一遍SPFA求去的最短路径，把图的边逆向，再一遍SPFA，求出回来的最短路径...
SPFA其实是Bellman-Ford使用队列的优化

  1 #include < iostream >
  2 #include < queue >
  3 using   namespace  std;
  4 #define  INF 2147483647    // 2,147,483,647
  5 #define  MAXP 1000000     // 1,000,000
  6 #define  MAXSUM 1000000000   // 1,000,000,000
  7
  8 struct  Edge
  9 {
 10    //int src;
 11    int vertex;
 12    int weight;
 13    Edge *next;
 14} ;
 15
 16 Edge edge[MAXP * 2 + 1 ];
 17 Edge  * go[MAXP + 1 ];
 18 Edge  * back[MAXP + 1 ];
 19 int  dis[MAXP + 1 ];
 20 bool  flag[MAXP + 1 ];   // false denote that the vertex is not in the queue
 21
 22 void  creatlist( int  vn, int  en)
 23 {
 24    int s,d,w,i,j;
 25    for(i=1;i<=vn;i++)
 26    {
 27        go[i]=NULL;
 28        back[i]=NULL;
 29        flag[i]=false;  
 30    }
 31    for(i=1,j=1;i<=en;i++)
 32    {
 33        scanf("%d%d%d",&s,&d,&w);
 34        edge[j].vertex=d;
 35        edge[j].weight=w;
 36        edge[j].next=go[s];
 37        go[s]=&edge[j++];
 38        
 39        edge[j].vertex=s;
 40        edge[j].weight=w;
 41        edge[j].next=back[d];
 42        back[d]=&edge[j++];
 43    }
 44}
 45
 46 void  spfa(Edge  * adjlist[], int  vn, int  src)
 47 {
 48    int i,u,v,tmp;
 49    Edge *p;
 50    queue<int> Queue;
 51    for(i=1;i<=vn;i++)
 52        dis[i]=INF;
 53    dis[src]=0;
 54    Queue.push(src);
 55    while(!Queue.empty())
 56    {
 57        u=Queue.front();
 58        Queue.pop();
 59        flag[u]=false;
 60        p=adjlist[u];
 61        while(p!=NULL)
 62        {
 63            tmp=dis[u]+p->weight;
 64            
 65            v=p->vertex;
 66            if(dis[v] > tmp)
 67            {
 68                dis[v]=tmp;
 69                if(flag[v]==false)
 70                {
 71                    Queue.push(v);
 72                    flag[v]=true;
 73                }
 74            }
 75            /**//*
 76            if(dis[p->vertex] > tmp)
 77            {
 78                dis[p->vertex]=tmp;
 79                if(flag[p->vertex]==false)
 80                {
 81                    Queue.push(p->vertex);
 82                    flag[p->vertex]=true;
 83                }
 84            }*/
 85            p=p->next;
 86        }
 87    }
 88}
 89
 90 int  main()
 91 {
 92    int n,p,q,i;
 93    __int64 sum;
 94    scanf("%d",&n);
 95    while(n--)
 96    {
 97        scanf("%d%d",&p,&q);
 98        creatlist(p,q);
 99        sum=0;
100        spfa(go,p,1);
101        for(i=1;i<=p;i++)
102            sum+=dis[i];
103        spfa(back,p,1);
104        for(i=1;i<=p;i++)
105            sum+=dis[i];
106        printf("%I64d\n",sum);
107    }
108    return 0;
109}