魔法宝石 spfa

1262: 魔法宝石

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

题目描述

小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

spfa算法

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f
#define spot 100000
#define edge 1000000
struct stu
{
    int x,y,z,ha;
} a[2*edge+10];
int b1[100005];
int dis[spot+10],first[spot+10],next1[2*edge+10],b[spot+10];
int main()
{
    queue<int>q;
    int t;
    scanf("%d",&t);
    int i,j,xx,yy,zz,k;
    while(t--)
    {
        int n,m;
        j=0;
        memset(dis,inf,sizeof(dis));
        memset(first,-1,sizeof(first));
        memset(b,0,sizeof(b));
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&b1[i]);
            dis[i]=b1[i];
        }
        for(int i=1; i<=m; i++)
        {
            scanf("%d%d%d",&xx,&yy,&zz);
            a[++j].x=xx,a[j].y=yy,a[j].z=zz;
            next1[j]=first[a[j].x];
            first[a[j].x]=j;
            a[++j].x=yy,a[j].y=xx,a[j].z=zz;
            next1[j]=first[a[j].x];
            first[a[j].x]=j;
            if(b[xx]==0)
            {
                q.push(xx);
                b[xx]=1;
            }
            if(b[yy]==0)
            {
                q.push(yy);
                b[yy]=1;
            }
        }
        while(!q.empty())
        {
            k=q.front();
            q.pop();
            b[k]=0;
            for(k=first[k]; k+1; k=next1[k])
            {
                if(dis[a[k].z]>dis[a[k].x]+dis[a[k].y])
                {
                    dis[a[k].z]=dis[a[k].x]+dis[a[k].y];
                    if(!b[a[k].z])
                        q.push(a[k].z),b[a[k].z]=1;
                }
            }
        }
        for(int i=1; i<=n; i++)
        {
            if(i<=n-1)
                printf("%d ",dis[i]);
            else
                printf("%d\n",dis[i]);
        }
    }
}