poj 1511(spfa)

---恢复内容开始---

http://poj.org/problem?id=1511

一个spfa类的模板水题。

题意：就是求从1到n个点的来回的所有距离和。

对spfa类的题还是不太熟练，感觉还是做少了，多水水这种题。

思路：也就是双向的spfa就行了。这里就是注意答案要用long long 类型来存就可以。

#include <stdio.h>
#include <string.h>
#include <queue>
#define maxn 1000010
#define inf 0x3f

using namespace std;

int m,n,pos,a[ maxn ],b[ maxn ],c[ maxn ],head[ maxn ];

long long ans , dist[ maxn ];

bool vis[ maxn ];

struct note {
    int v,w,next;
}edge[maxn];

void init()
{
    pos = 1;
    memset( dist , inf , sizeof( dist ) );
    memset( head , -1 , sizeof( head ) );
    memset( vis , false ,sizeof( vis ) );
}

void add(int x,int v,int w)    //这是用数组来构建的一个邻接表。不懂可以在纸上模拟一次就行。
{
    edge[ pos ].v = v;
    edge[ pos ].w = w;
    edge[ pos ].next = head[ x ];
    head[ x ] = pos++;
}

void spfa()     //标准的spfa模板。如果可能有负权回路的话，那就加个num数组来判断。
{
    queue<int >s;
    s.push(1);
    vis[ 1 ] = true;
    dist[ 1 ] = 0;
    while(!s.empty())
    {
        int tmp = s.front();
        s.pop();
        vis [ tmp ] = false;
        for( int i = head[ tmp ] ; i != -1 ; i = edge[ i ].next )
        {
            if( dist[ edge[ i ].v ] > dist[ tmp ] + edge[ i ].w)
            {
                dist[ edge[ i ].v ] = dist[ tmp ] + edge[ i ].w;
                if( !vis[ edge[ i ].v ] )
                {
                    s.push( edge[ i ].v );
                    vis[ edge[ i ].v ] =true;
                }

            }

        }
    }
    for( int i = 1 ; i <= m ; i++ )
        ans += dist[ i ];
}

int main()
{
  //  freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        ans = 0;
        scanf("%d%d",&m,&n);
        init();              //两次spfa，正向和反向。
        for( int i = 1 ; i <= n ; i++ )
        {
            scanf("%d%d%d",&a[ i ],&b[ i ],&c[ i ]);
            add( a[ i ] , b[ i ] , c[ i ] );
        }
        spfa();
        init();
        for( int i = 1 ; i <= n ; i++ )
            add( b[ i ] , a[ i ] , c[ i ] );
        spfa();
        printf("%lld
",ans);
    }
    return 0;
}