题目描述
小s想要创造n种魔法宝石。小s可以用ai的魔力值创造一棵第i种魔法宝石,或是使用两个宝石合成另一种宝石(不消耗魔力值)。请你帮小s算出合成某种宝石的所需的最小花费。
输入
第一行为数据组数T(1≤T≤3)。
对于每组数据,首先一行为n,m(1≤n,m≤10^5)。分别表示魔法宝石种类数和合成魔法的数量。
之后一行n个数表示a1到an。(1≤ai≤10^9)。a_i表示合成第i种宝石所需的魔力值。
之后n行,每行三个数a,b,c(1≤a,b,c≤n),表示一个第a种宝石和第b种宝石,可以合成一个第c种宝石。
输出
每组数据输出一行n个数,其中第i个数表示合成第i种宝石的魔力值最小花费。
样例输入
1
3 1
1 1 10
1 2 3
样例输出
1 1 2
用vector存图 #include<stdio.h> #include<string.h> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<math.h> #include<vector> using namespace std; struct T { int x,y,z; }t[100005]; long long a[100005]; vector<T>sh[100005]; T temp; void dfs(int z) { for(int i=0;i<sh[z].size();i++) { temp=sh[z][i]; if(a[temp.z]>a[temp.x]+a[temp.y]) { a[temp.z]=a[temp.x]+a[temp.y]; dfs(temp.z); } } } int main() { int tt; scanf("%d",&tt); while(tt--) { memset(a,0,sizeof(a)); memset(t,0,sizeof(t)); memset(sh,0,sizeof(sh)); int n,m,l,p,j; scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%lld",&a[i]); } for(int i=0;i<m;i++) { int x1,y1,z1; scanf("%d%d%d",&x1,&y1,&z1); t[i].x=x1; t[i].y=y1; t[i].z=z1; temp=t[i]; sh[x1].push_back(temp); temp.x=y1; temp.y=x1; sh[y1].push_back(temp); } for(int i=0;i<m;i++) { l=t[i].x,p=t[i].y,j=t[i].z; if(a[j]>a[l]+a[p]) { a[j]=a[l]+a[p]; dfs(j); } } printf("%lld",a[1]); for(int i=2;i<=n;i++) { printf(" %lld",a[i]); } printf(" "); } }