计蒜客—圣诞树

解题说明：稍等

AC代码：

#include<iostream>
#include<vector>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
int mark[50005],cow[50005];
ll dis[50005];
int n,m;
struct po{
	int d;int w;
	po(int dd,int ww){d=dd;w=ww;}
};
vector<po>ve[50005];
set<int>st; 
void dij(){
	int now=1;
	for(int i=1;i<n;i++){
		for(int j=0;j<ve[now].size();j++){
              if(mark[ve[now][j].d])continue;
			  if(dis[ve[now][j].d]>dis[now]+ve[now][j].w)
			    dis[ve[now][j].d]=dis[now]+ve[now][j].w;
		}
		int mn=inf;
		for(int k=2;k<=n;k++){
			if(mark[k])continue;
			if(dis[k]<mn){
				mn=dis[k];
				now=k;
			}
		}
		mark[now]=1;
		st.insert(now);
	}
	return ;
}
int main(){
	cin>>n>>m;
	memset(dis,inf,sizeof(dis));
	memset(mark,0,sizeof(mark));
	for(int i=1;i<=n;i++){
		cin>>cow[i];
	}
	for(int i=1;i<=m;i++){
		int ip1,ip2,ip3;
		cin>>ip1>>ip2>>ip3;
		ve[ip1].push_back(po(ip2,ip3));
		ve[ip2].push_back(po(ip1,ip3));
	}
	st.insert(1); 
	mark[1]=1;
	dis[1]=0;
	dij();
	ll sum=0;
	if(st.size()!=n)cout<<"No Answer\n";
	else{
        for(int i=2;i<=n;i++)
		   sum+=dis[i]*cow[i];
	       cout<<sum<<endl;
    }
	return 0;
}