Pagini recente » Cod sursa (job #883863) | Cod sursa (job #217424) | Cod sursa (job #2384087) | Cod sursa (job #1874584) | Cod sursa (job #2519552)
#include <fstream>
#include <queue>
#include <vector>
#define N 50002
#define INF 2000000000
using namespace std;
int dc[N],dist[N];
struct bla
{
int no,co;
};
vector <bla> graph[N];
struct how
{
bool operator()(bla A, bla B)
{
return (A.co>B.co);
}
};
priority_queue <bla, vector <bla>, how> q;
bool viz[N];
void dijkstra(int nod)
{
int ve,d;
viz[nod]=1;
dist[nod]=0;
q.push({nod,0});
while(!q.empty())
{
nod=q.top().no;
d=q.top().co;
q.pop();
viz[nod]=0;
for(int i=0;i<graph[nod].size();++i)
{
ve=graph[nod][i].no;
if(!viz[ve])
if(dist[ve]>dist[nod]+d)
viz[ve]=1,dist[ve]=dist[nod]+d,q.push({ve,dist[ve]});
}
}
}
int main()
{
ifstream f("distante.in");
ofstream g("distante.out");
int t,n,m,st,x,y,c,ok;
f>>t;
while(t--)
{
f>>n>>m>>st;
ok=1;
for(int i=1;i<=n;++i) f>>dc[i],dist[i]=INF,viz[i]=0;
for(int i=1;i<=m;++i)
{
f>>x>>y>>c;
graph[x].push_back({y,c});
graph[y].push_back({x,c});
}
dijkstra(st);
for(int i=1;i<=n;++i)
if(dist[i]!=dc[i])
{
g<<"NU\n";
ok=0;
break;
}
if(ok)
g<<"DA\n";
for(int i=1;i<=n;++i) graph[i].clear();
}
f.close();
g.close();
return 0;
}