Pagini recente » Cod sursa (job #1451149) | Cod sursa (job #3038675) | Cod sursa (job #2069899) | Cod sursa (job #2982292) | Cod sursa (job #2087330)
#include <cstdio>
#include <vector>
#include <queue>
#define INF 99999999
using namespace std;
int T,n,m,DC[50003],D[50003],start;
vector < pair < int , int > > L[50002];
priority_queue < pair < int , int > > Heap;
bool Dijkstra(int Nstart)
{
bool Da=true;
int node,cost,neigh,son,sum;
for(int i=1;i<=n;i++)
D[i] = INF;
D[Nstart] = 0;
Heap.push({0,Nstart});
while(!Heap.empty())
{
cost = -Heap.top().first;
node = Heap.top().second;
Heap.pop();
if(cost <= D[node])
{
neigh = L[node].size();
for(int i=0;i<neigh;i++)
{
son = L[node][i].first;
sum = L[node][i].second;
if(D[son] > sum + cost)
{
D[son] = sum + cost;
Heap.push({-D[son],son});
}
}
}
}
for(int i=1;i<=n;i++)
{
if(DC[i] != D[i])
{
return false;
}
}
return true;
}
void Read()
{
int a,b,c;
freopen("distante.in","r",stdin);
freopen("distante.out","w",stdout);
scanf("%i",&T);
for(int k=1;k<=T;k++)
{
scanf("%i %i %i",&n,&m,&start);
for(int i=1;i<=n;i++)
scanf("%i",&DC[i]);
for(int i=1;i<=m;i++)
{
scanf("%i %i %i",&a,&b,&c);
L[a].push_back({b,c});
L[b].push_back({a,c});
}
if(Dijkstra(start) == true)
printf("DA\n");
else
printf("NU\n");
for(int i=1;i<=n;i++)
D[i] = INF;
}
}
int main()
{
Read();
return 0;
}