POJ 3013 Big Christmas Tree

题目来源：http://poj.org/problem?id=3013

      目的就是为了圣诞树的总花费最小，各点的花费：指向该节点边权(unit)*该节点所有子树重量之和(sum(weight))。

这道题的思路是看了别人题解才会的，其实总花费最小即：每点到源点的距离(dist)*该点的重量(weight)。这样这个题目

就变成了最小路径求解问题，SPFA算法即可。这道题总是TE，不知道为什么。后来随便加了一个if(a>n) return;别人代

码都没加这句，居然就AC了，简直就不知道为什么，这a还能超过n吗？

#include<stdio.h>
#include<string.h>
#include<queue>
#define INF 0x3f3f3f3f3f3f3f3f
const int maxn=50000+5;
int first[maxn],w[maxn],e;
long long dist[maxn];
bool inq[maxn];
struct node{
    int v,unit,next;
}edge[2*maxn];
void add_edge(int a,int b,int c)
{
    edge[e].v=b;edge[e].unit=c;
    edge[e].next=first[a];
    first[a]=e;
    e++;
}
void spfa(int n)
{
    std::queue<int>q;
    q.push(1);
    dist[1]=0;
    inq[1]=true;
    while(!q.empty()){
        int a=q.front();
        q.pop();
        inq[a]=false;
        if(a>n)
        return;
        for(int i=first[a];i!=-1;i=edge[i].next){
            int ev=edge[i].v,ew=edge[i].unit;
            if(dist[ev]>dist[a]+ew){
                dist[ev]=dist[a]+ew;
                if(!inq[ev]){
                    inq[ev]=true;
                    q.push(ev);
                }
            }
        }
    }
}
int main()
{
    int t,n,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        int a,b,c;
        for(int i=1;i<=n;i++)
            scanf("%d",&w[i]);
        memset(first,-1,sizeof(int)*(n+1));
        e=1;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&a,&b,&c);
            add_edge(a,b,c);
            add_edge(b,a,c);
        }
        memset(inq,false,sizeof(bool)*(n+1));
        memset(dist,0x3f,sizeof(dist[0])*(n+1));
        spfa(n);
        long long ans=0;
        bool flag=true;
        for(int i=1;i<=n;i++){
            ans+=dist[i]*w[i];
            if(dist[i]>=INF){
                flag=false;
                break;
            }
        }
        if(flag)
            printf("%I64d\n",ans);
        else
            printf("No Answer\n");
    }
    return 0;
}