Pagini recente » Cod sursa (job #1418152) | Cod sursa (job #2373288) | Cod sursa (job #751412) | Cod sursa (job #312686) | Cod sursa (job #2336336)
#include <iostream>
#include <fstream>
#include <set>
#include <vector>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
set < pair<int,int> >s;
vector <pair<int,int> >L[60000];
int d[60000],d2[60000];
int n,m,gr,g,i,x,y;
int nod,cost,vecin,start;
bool verif(int a[],int b[],int n){
for(int i=1;i<=n;i++)
if(a[i]!=b[i])
return 0;
return 1;
}
int main()
{
int t;
fin>>t;
for(; t; t--)
{
fin>>n>>m>>start;
for(i=1;i<=n;i++)
L[i].clear();
for(i=1;i<=n;i++)
fin>>d2[i];
for(i=1; i<=m; i++)
{
fin>>x>>y>>gr;
L[x].push_back({y,gr});
L[y].push_back({x,gr});
}
for(i=1; i<=n; i++)
d[i]=2000000000;
d[start]=0;
s.insert({0,start});
while(s.empty()==0)
{
nod=s.begin()->second;
s.erase(s.begin());
for(i=0; i<L[nod].size(); i++)
{
cost=L[nod][i].second;
vecin=L[nod][i].first;
if(d[nod]+cost<d[vecin])
{
s.erase({d[vecin],vecin});
d[vecin]=cost+d[nod];
s.insert({d[vecin],vecin});
}
}
}
if(verif(d,d2,n)==1)
fout<<"DA\n";
else fout<<"NU\n";
}
}