最短路SPFA——Big Christmas Tree ( POJ 3013 )

• 题目链接：
  http://poj.org/problem?id=3013

• 分析：
  给出N个点和M条边，每个点有它的权值，每条边也有它的权值，建造这条边的花费等于这条边的权值*另一端端点的子树所包含的点的权值和。求连接所有点以点1为根节点构造出一颗树的最小花费。

• 题解：
  通过计算可以推断出，每个点的权值会被计算多次，但是只有从根节点到该点的边才会乘以该点的权值，所以我们可以把问题转换为求根节点1到所有点的最短路，每条路乘以点的权重即得到整棵树的花费。

• AC代码：
  

/*************************************************************************
    > File Name: test.cpp
    > Author: Akira 
    > Mail: qaq.febr2.qaq@gmail.com 
 ************************************************************************/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <set>
#include <list>
#include <ctime>
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))
using namespace std;

#define MaxN 66666
#define MaxM MaxN*10
#define INF 20000000000
#define bug cout<<88888888<<endl;

int N,M;
int A[MaxN];

struct Edge
{
    int u,v;
    int cost;
    int next;
}edge[MaxM];


int head[MaxN];
int cont;

void add(int u, int v, int cost)
{
    edge[cont].v = v;
    edge[cont].u = u;
    edge[cont].cost = cost;
    edge[cont].next = head[u];
    head[u] = cont++;
}

void Add(int u, int v, int cost)   //添加u-v，权重为cost的边，这个建的是双向，可以改
{
    add(u, v, cost);  
    add(v, u, cost);   
}

void init()                 //!!!!!!!!每次建图前记得init!!!
{
    cont = 0;
    MST(head,-1);
}

bool vis[MaxN];
LL dis[MaxN];

LL SPFA(int s, int t)
{
    int n = MaxN;         //n赋值为图中点数
    int k, p, V;
    queue<int> q;
    CLR(vis);
    for(int i=1;i<=n;i++)
        dis[i] = INF;
    q.push(s);
    vis[s] = 1;
    dis[s] = 0;
    while(!q.empty())
    {
        k = q.front();
        q.pop();
        vis[k] = 0;
        for(p = head[k]; p!=-1; p = edge[p].next)
        {
            V = edge[p].v;
            if(edge[p].cost + dis[k] < dis[V])
            {
                dis[V] = edge[p].cost + dis[k];
                if(!vis[V])
                {
                    q.push(V);
                    vis[V] = 1;
                }
            }
        }
    }
    if(dis[t] == INF) return -1;   //表示有负权
    return dis[t];
}



int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        init();
        scanf("%d%d", &N, &M);
        for(int i=1;i<=N;i++)
            scanf("%d", &A[i]);
        int a,b,c;
        for(int i=0;i<M;i++)
        {
            scanf("%d%d%d", &a, &b, &c);
            Add(a,b,c);
        }


        LL Ans = 0;
        SPFA(1,N);


        int flag = 0;
        for(int i=2;i<=N;i++)
        {
            if(dis[i] == INF) //表示有点不可达
            {
                flag = 1;
                break;
            }
            Ans += A[i]*dis[i];
        }

        if(flag)
        {
            cout << "No Answer\n";
            continue;
        }
        cout << Ans << endl;
    }
    system("pause");
}