poj3013Big Chrismas Tree——树转换spfa

Posted 2020-10-27 Zinn

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

看似生成树，实则最短路，可以将题意转化为点权*根到此点的边权和(最短路使其最小)。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
queue<int>q;
int t,n,m,head[50005],ct,s[50005];
bool in[50005];
ll dis[50005],INF=1e10,ans;
struct N{
    int to,next,w;
    N(int t=0,int n=0,int ww=0):to(t),next(n),w(ww) {}
}edge[100005];
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        ct=0;
        memset(head,0,sizeof head);
        memset(in,0,sizeof in);
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%d",&s[i]),dis[i]=INF;
        for(int i=1;i<=m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            edge[++ct]=N(y,head[x],z);head[x]=ct;
            edge[++ct]=N(x,head[y],z);head[y]=ct;
        }
        while(q.size())q.pop();
        dis[1]=0;q.push(1);in[1]=1;
        while(q.size())
        {
            int x=q.front();q.pop();
            in[x]=0;
            for(int i=head[x];i;i=edge[i].next)
            {
                int u=edge[i].to;
                if(dis[u]>dis[x]+edge[i].w)
                {
                    dis[u]=dis[x]+edge[i].w;
                    if(!in[u])in[u]=1,q.push(u);
                }
            }
        }
        bool flag=0;ans=0;
        for(int i=1;i<=n;i++)
        {
            if(dis[i]==INF)
            {
                flag=1;break;
            }
            ans+=dis[i]*s[i];
        }
        if(flag)printf("No Answer\n");
        else printf("%lld\n",ans);
    }
    return 0;
}