Pagini recente » Cod sursa (job #2630461) | Cod sursa (job #507912) | Cod sursa (job #2394264) | Cod sursa (job #3246186) | Cod sursa (job #1518682)
#include <iostream>
#include <fstream>
#include <set>
#include <vector>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
vector<int> v[1000005],c[1000007];
set<pair< int, int> >s;
int d[1000005],n,m,inf=10000000000;
int ok[1000005];
void dijkstra(int nod)
{
int x,val;
for(int i=1;i<=n;i++)d[i]=inf;
d[nod]=0;
s.insert(make_pair(0,nod));
while(s.size()>0)
{
val=(*s.begin()).first;
x=(*s.begin()).second;
s.erase(*s.begin());
for(int i=0;i<v[x].size();i++)
{
if(d[ v[x][i]] > c[x][i]+val)
{
d[v[x][i]]=c[x][i]+val;
s.insert(make_pair(d[v[x][i]],v[x][i]));
}
}
}
}
int main()
{
int x,y,z,t,nod;
fin>>t;
for(int j=1;j<=t;j++)
{
fin>>n>>m>>nod;
for(int i=1;i<=n;i++)
fin>>ok[i];
for(int i=1;i<=m;i++)
{
fin>>x>>y>>z;
v[x].push_back(y);
c[x].push_back(z);
}
dijkstra(nod);
int k=0;
for(int i=1;i<=n;i++)
if(ok[i]!=d[i])
k++;
if(k!=0)fout<<"NU\n";
else fout<<"DA\n";
}
}