Pagini recente » Cod sursa (job #1702627) | Cod sursa (job #1003828) | Cod sursa (job #21878) | Cod sursa (job #371493) | Cod sursa (job #2884509)
#include <iostream>
#include <fstream>
#include <vector>
#include <set>
#define INF 2147483647
using namespace std;
ifstream fi("distante.in");
ofstream fo("distante.out");
int T;
vector <int>BRZ;
vector <pair<int,int> >V[50001];
set <pair<int,int> >S;
int D[50001];
void Dijkstra(int source)
{
while(!S.empty())
{
int vertex=S.begin()->second;
S.erase(S.begin());
for(int i=0;i<V[vertex].size();i++)
{
int vecin=V[vertex][i].first;
if(vecin!=source)
if(D[vecin]>D[vertex]+V[vertex][i].second)
{
D[vecin]=D[vertex]+V[vertex][i].second;
S.insert(make_pair(D[vecin],vecin));
}
}
}
}
int main()
{
fi>>T;
for(int test=1;test<=T;test++)
{
int n,m,source;
fi>>n>>m>>source;
BRZ.clear();
for(int i=1;i<=n;i++)
{
D[i]=INF;
V[i].clear();
int x;
fi>>x;
BRZ.push_back(x);
}
for(int i=1;i<=m;i++)
{
int nodeX,nodeY,C;
fi>>nodeX>>nodeY>>C;
V[nodeX].push_back(make_pair(nodeY,C));
V[nodeY].push_back(make_pair(nodeX,C));
}
D[source]=0;
S.insert(make_pair(0,source));
Dijkstra(source);
int GG=0;
for(int i=1;i<=n && GG==0;i++)
{
if(D[i]!=BRZ[i-1])
{
GG=1;
continue;
}
}
if(!GG)
fo<<"DA\n";
else
fo<<"NU\n";
}
return 0;
}