3107 小明爱宝石
小明想要创造n种魔法宝石。小明可以用ai的魔力值创造一棵第i种魔法宝石,或是使用两个宝石合成另一种宝石(不消耗魔力值)。请你帮小明算出合成某种宝石的所需的最小花费。
输入
首先一行为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种宝石的魔力值最小花费。
数据范围
对于10%的数据,1 <= n <= 8,1 <= m<= 100000,1≤a,b,c≤n
对于50%的数据,1 <= n <= 1024,1 <= m<= 100000,1≤a,b,c≤n
对于100%的数据,1 <= n <= 100000,1 <= m<= 100000,1≤a,b,c≤n
输入样例
3 1
1 1 10
1 2 3
输出样例
1 1 2
样例解释
第一个宝石花费1,第二个宝石花费1,第三个宝石可以由第一个第二个宝石合成花费2。
解析:
我们可以进行bfs,设C[x]为合成x宝石的最少花费,那么初始C[x]为给定不用合成需要的花费,根据下面给定的关系,我们把上面的C数组放进优先队列内,每次取出花费最少的值进行扩展,扩展规则是它下面给的花费规则,判断是否可以降低c宝石的花费,时间复杂度为O(mlogm)。然后观察上面的求解过程,你发现这个就是一个最短路,给定初始权值,下面的关系相当于地图,然后直接跑堆优化的dijkstra即可。
放代码:
#include<bits/stdc++.h>
using namespace std;
struct node
{
int x,y,z;
}b[100000+5];
int main()
{
int a[100000+5];
int i,j;
bool cnt;
int n,m;
cin>>n>>m;
for(i=1;i<=n;i++)
cin>>a[i];
for(i=1;i<=m;i++)
cin>>b[i].x>>b[i].y>>b[i].z;
while(1)
{
cnt=false;
for(i=1;i<=m;i++){
if(a[b[i].x]+a[b[i].y]<a[b[i].z]){
a[b[i].z]=a[b[i].x]+a[b[i].y];
cnt=true;
}
}
if(!cnt)
break;
}
cout<<a[1];
for(i=2;i<=n;i++){
cout<<" "<<a[i];
}
cout<<endl;
return 0;
}