魔法宝石

来源:互联网 发布:制作招生简章的软件 编辑:程序博客网 时间:2024/04/28 23:26

题目描述

小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,cn),表示一个第a种宝石和第b种宝石,可以合成一个第c种宝石。

输出

每组数据输出一行n个数,其中第i个数表示合成第i种宝石的魔力值最小花费。

样例输入

1
3 1
1 1 10
1 2 3

样例输出

1 1 2
本题回溯,如果合成的宝石花费会减小那么就不断往回更新以该宝石为材料能合成的宝石
#include<bits/stdc++.h>using namespace std;int b[100010];vector<int>a[100010];//a[i]保存以i为材料的魔法,偶数是另一个作为材料得宝石,奇数是合成的宝石void ss(int C)//更新以C为材料的宝石{    for(int i=0;i<a[C].size();i+=2)    {        int h=a[C][i],j=a[C][i+1];        if(b[j]>b[h]+b[C])//判断是否应该更新        {            b[j]=b[h]+b[C];            //printf("%d*\n",j);            ss(j);        }    }    return;}int main(){    int T;    scanf("%d",&T);    while(T--)    {         int n,m;         scanf("%d%d",&n,&m);         for(int i=1;i<=n;i++)            scanf("%d",&b[i]);         while(m--)         {             int A,B,C;             scanf("%d%d%d",&A,&B,&C);             a[A].push_back(B);//保存魔法             a[A].push_back(C);             a[B].push_back(A);             a[B].push_back(C);             if(b[C]>b[A]+b[B])             {                 b[C]=b[A]+b[B];                 for(int i=0;i<a[C].size();i+=2)                {                    int h=a[C][i],j=a[C][i+1];                    if(b[j]>b[h]+b[C])                    {                        b[j]=b[h]+b[C];                        ss(j);                    }                }             }         }         printf("%d",b[1]);         a[1].clear();         for(int i=2;i<=n;i++)         printf(" %d",b[i]),a[i].clear();//清空动态数组         printf("\n");    }}


0 0
原创粉丝点击