Pagini recente » Cod sursa (job #1836342) | Cod sursa (job #1932081) | Cod sursa (job #2158329) | Cod sursa (job #2850383) | Cod sursa (job #2323654)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <functional>
#include <string.h>
#define pii pair < int , int >
#define INF 0x3f3f3f3f
std::ifstream in("distante.in");
std::ofstream out("distante.out");
using namespace std;
priority_queue < pii , vector <pii> , greater < pii > > pq;
vector < pair <int , int > > v[50005];
int costurile_initiale[50005];
int costurile_reale[50005];
int n,m,start,t;
void dijkstra(int start)
{
pq.push({costurile_initiale[start],start});
while(!pq.empty())
{
int nod_curent = pq.top().second;
int cost_curent = pq.top().first;
pq.pop();
if(costurile_reale[nod_curent]!=INF)
{
for(int i=0;i<v[nod_curent].size();i++)
{
int nod_vecin = v[nod_curent][i].first;
int cost_vecin = v[nod_curent][i].second;
if(costurile_reale[nod_curent] + cost_vecin < costurile_reale[nod_vecin])
{
costurile_reale[nod_vecin]=costurile_reale[nod_curent] + cost_vecin;
pq.push({costurile_reale[nod_vecin],nod_vecin});
}
}
}
}
}
int main()
{
int x,y,cost;
in>>t;
for(int teste=1;teste<=t;teste++)
{
in>>n>>m>>start;
for(int i=1;i<=n;i++)
{
in>>costurile_initiale[i];
costurile_reale[i]=INF;
v[i].clear();
}
costurile_reale[start]=0;
for(int i=1;i<=m;i++)
{
in>>x>>y>>cost;
v[x].push_back({y,cost});
v[y].push_back({x,cost});
}
dijkstra(start);
bool ebun=1;
for(int i=1;i<=n;i++)
{
if(costurile_initiale[i]!=costurile_reale[i])
{
ebun=0;
break;
}
}
if(ebun)
{
out<<"DA\n";
}
else out<<"NU\n";
}
return 0;
}