POJ 3013 Big Christmas Tree（SPFA）

題目鏈接：

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

題意：

給你一顆樹，每棵子樹的造價 = 子樹上所有節點的和 * 子樹和父親節點的權值。求最小造價，如果不能造出來輸出No Answer。

題解：

其實這顆樹的造價就是源點1到每個點的距離與每個點的造價的乘積。故可以轉化成最短路求解，求單源最短路可以使用SPFA求。

AC代碼：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#define MAXN 60000

using namespace std;

long long dist[MAXN],head[MAXN];
bool vis[MAXN];
int cnt,n,m;
int weight[MAXN];
const unsigned long long INF = 9999999999LL;
queue <int> Q;

struct node
{
    int u,v,w,next;
}edge[MAXN * 2];

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

void spfa(int s)
{
    for(int i=0; i<=n; i++)
    {
        dist[i]=INF;
        vis[i]=false;
    }
    dist[s]=0;
    Q.push(s);
    vis[s]=true;
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            if(dist[u]+edge[i].w<dist[v])
            {
                dist[v]=dist[u]+edge[i].w;
                if(!vis[v])
                {
                    vis[v]=true;
                    Q.push(v);
                }
            }
        }
    }
}

void init()
{
    while(!Q.empty())
    {
        Q.pop();
    }
    cnt=0;
    memset(head, -1, sizeof(head));
}

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        init();
        scanf("%d %d", &n, &m);
        for(int i = 1; i <= n; i++)
            scanf("%d", &weight[i]);
        while(m--)
        {
            int a,b,c;
            scanf("%d %d %d", &a, &b, &c);
            add(a,b,c);
        }
        spfa(1);
        int flag = 0;
        long long ans = 0;
        for(int i = 1; i <= n; i++)
        {
            if(dist[i] == INF)
            {
                flag = 1;
                break;
            }
            ans += dist[i] * weight[i];
        }
        if(!flag)
            printf("%I64d\n", ans);
        else
            printf("No Answer\n");
    }
    return 0;
}