hautoj 1262 魔法宝石【最短路】

点击打开链接

时间限制: 2 秒  内存限制: 64 MB
提交: 510  解决: 150
提交  状态 

题目描述

小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种宝石的魔力值最小花费。

样例输入

13 11 1 101 2 3

样例输出

1 1 2

当时看出来了，但是没写，因为简单题还没写出来呢，爆炸。

就是个最短路 。所有能合成的都跑一边就行了。

#include<bits/stdc++.h>

#define ll long long
using namespace std;

const int maxn=1e5+10;
int T,n,m,cnt;
struct node{
    int u,to;
    int nxt;
}e[maxn*2];
int head[maxn];
void addedge(int a,int b,int v){
    e[++cnt].to=v;
    e[cnt].u=b;
    e[cnt].nxt=head[a];
    head[a]=cnt;
}
bool vis[maxn];
int dis[maxn];
int main(){
    scanf("%d",&T);
    int a,b,c;
    while(T--){
        cnt=1;
        memset(vis,0,sizeof(vis));
        memset(head,-1,sizeof(head));
        scanf("%d %d",&n,&m);
        for(int i=1;i<=n;++i) scanf("%d",&dis[i]);
        for(int i=1;i<=m;++i){
            scanf("%d %d %d",&a,&b,&c);
            addedge(a,b,c);
            addedge(b,a,c);
        }
        queue<int>que;
        while(!que.empty()) que.pop();
        for(int i=1;i<=n;++i) que.push(i);
        while(!que.empty()){
            int u=que.front();
            que.pop();
            vis[u]=false;
            for(int i=head[u];i!=-1;i=e[i].nxt){
                int v=e[i].to;
                if(dis[v]>dis[u]+dis[e[i].u]){
                    dis[v]=dis[u]+dis[e[i].u];
                    if(!vis[v]){
                        vis[v]=true;
                        que.push(v);
                    }
                }
            }
        }
        for(int i=1;i<n;++i) printf("%d ",dis[i]);
        printf("%d\n",dis[n]);
    }
    return 0;
}