Big Christmas Tree POJ - 3013 一种最短路的出题姿势

题意：给n个点从1到n标号，下面一行是每个点的权，另外给出m条边，下面是每条边的信息，两个端点+权值，边是无向边。你的任务是选出一些边，使这个图变成一棵树。这棵树的花费是这样算的，1号固定为树根，树中每个父节点下面的边都有个边权，然后边权乘上这条边的下面所有的子孙后代的点权和.

题目换一个角度就会发现其实就是一个裸的最短路,即答案也是每个点权乘上从起点到终点的边权和,即是最短路

代码如下:

/*
 * Author       :  Echo
 * Email        :  1666424499@qq.com  
 * Description  :   
 * Created Time :  2017/10/6 9:26:12
 * Last Modify  :  2017/10/6 10:11:48
 * File Name    :  a.cpp
 */
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <string>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <time.h>
#define LL long long
#define mem(a,k) memset(a,k,sizeof(a))
using namespace std;
const int maxint = -1u>>1;
const int maxn=1e5;
const int maxm=1e6;
bool vis[maxn];
long long dis[maxn];
long long bn[maxn];
struct edge{
    int to,next;
    long long value;
    void set(int a,long long b,int c){
        to=a,value=b;next=c;
    }
}an[maxm];
int cnt=0,head[maxn];
void addedge(int u,int v,long long value){
    an[cnt].set(v,value,head[u]);
    head[u]=cnt++;
    an[cnt].set(u,value,head[v]);
    head[v]=cnt++;
}
void SPFA(int n){
    for(int i=1;i<=n;i++){
        vis[i]=0;
        dis[i]=1e17;
    }
    vis[1]=1;
    queue<int> que;
    dis[1]=0;
    que.push(1);
    while(!que.empty()){
        int u=que.front();
        que.pop();
        vis[u]=0;
        for(int i=head[u];i!=-1;i=an[i].next){
            int v=an[i].to;
            if(dis[v]>dis[u]+an[i].value){
                dis[v]=dis[u]+an[i].value;
                if(!vis[v])que.push(v);
                vis[v]=1;
            }
        }
    }
    long long ans=0;
    for(int i=1;i<=n;i++){
        if(dis[i]==1e17){
             printf("No Answer\n");return;
        }
        ans+=dis[i]*bn[i];
    }
    printf("%lld\n",ans);
}
int main(){
    int t;
    cin>>t;
    while(t--){
        memset(head,-1,sizeof(head));
        cnt=1;
        int n,m;
        cin>>n>>m;
        for(int i=1;i<=n;i++){
            scanf("%d",&bn[i]);
        }
        for(int i=1;i<=m;i++){
            int a,b;
            long long c;
            scanf("%d%d%lld",&a,&b,&c);
            addedge(a,b,c);
        }
        SPFA(n);
    }
    return 0;
}