hautoj1262 魔法宝石 DFS

#include <stdio.h>
#include <string.h>
#include <vector>
using namespace std;
struct node
{
int x,y,z;
}h[100005];


vector<node>link[100005];
long long a[100005];
node temp;


void dfs(int z)
{
for(int j=0;j<link[z].size();j++)
{
temp=link[z][j];
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 t;
scanf("%d",&t);
while(t--)
{
int n,m;
memset(link,0,sizeof(link));
memset(h,0,sizeof(h));
memset(a,0,sizeof(a));
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);//合成第i种宝石所需的魔力值 
int x,y,z;

for(int i=0;i<m;i++)
{
scanf("%d %d %d",&x,&y,&z);
h[i].x=x;h[i].y=y;h[i].z=z;
temp.x=x,temp.y=y,temp.z=z;

link[x].push_back(temp);
temp.x=y；
temp.y=x;
link[y].push_back(temp);
}

for(int i=0;i<m;i++)
{
x=h[i].x,y=h[i].y,z=h[i].z;
// printf("%d %d %d\n",x,y,z);
if(a[z]>a[x]+a[y])
{
a[z]=a[x]+a[y];
dfs(z);
}
}

printf("%lld",a[1]);
for(int i=2;i<=n;i++)
{
printf(" %lld",a[i]);
}
printf("\n");
}
return 0;
}