题解:poj3013 Big Christmas Tree 【最短路】

传送门

这个题题意是,给一个n个节点的图,在要求其一棵以1结点为根的生成树使树的边权和最小,树边权 = 对应的图边权 * 树边末端点为根的子树所有结点对于图顶点的点权和

然后就是直接跑最短路

(我爱SPFA)

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
#define N 500050
#define ll long long
const ll INF=1e15;

struct Edge {
    int next,v;
    ll w;
} e[N];

int h[N],cnt=0,n;
void addEdge(int a,int b,int w) {
    e[++cnt].next=h[a];
	e[cnt].v=b;
	e[cnt].w=w;
	h[a]=cnt;
}
#define QXX(u) for(int i=h[u],v;v=e[i].v,i;i=e[i].next)
ll dis[N],vis[N];
bool Spfa() {
    for(int i=1; i<=n; i++) 
        dis[i]=INF,vis[i]=0;
    dis[1]=0;
    vis[1]=1;
    queue<int>q;
    q.push(1);
    while(!q.empty()) {
        int u=q.front();
		q.pop();
        QXX(u) 
            if(dis[v]>dis[u]+e[i].w) {
                dis[v]=dis[u]+e[i].w;
                if(!vis[v]) 
					vis[v]=1,q.push(v);
            }
    }
    for(int i=1;i<=n;i++)
        if(dis[i]==INF)
		    return 0;
    return 1;
}

int verw[N];
int main() {
//	freopen("1.txt","r",stdin);
    int t;
    scanf("%d",&t);
    while(t--) {
        cnt=0;
        int en;
        scanf("%d%d",&n,&en);
        for(int i=1; i<=n; i++)
            scanf("%d",verw+i);
        memset(h,0,sizeof(h));
        for(int i=0; i<en; i++) {
            int a,b,w;
            scanf("%d%d%d",&a,&b,&w);
            addEdge(a,b,w);
            addEdge(b,a,w);
        }
        
        if(!Spfa()) {
            puts("No Answer");
            continue;
        }
        else {
		    ll res=0;
            for(int i=1; i<=n; i++)
                res+=verw[i]*dis[i];
            printf("%lld\n",res);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43464026/article/details/88356739