Pagini recente » Cod sursa (job #1147831) | Cod sursa (job #2239548) | Cod sursa (job #2351691) | Cod sursa (job #1082955) | Cod sursa (job #1855657)
#include <fstream>
#include <utility>
#include <vector>
#include <set>
#define inf 0x3f3f3f3f
#define nmax 50001
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
vector<pair<int,int> >::iterator it;
vector<pair<int,int> >a[nmax];
set<pair<int,int> >heap;
int cost[nmax],d[nmax];
int n,m,t,start;
void citire()
{
int i,x,y,c;
fin>>n>>m>>start;
for(i=1;i<=n;i++)
fin>>cost[i];
for(i=1;i<=m;i++)
{
fin>>x>>y>>c;
a[x].push_back(make_pair(y,c));
a[y].push_back(make_pair(x,c));
}
}
void dijkstra()
{
int i,nod,c,next;
for(i=1;i<=n;i++)
d[i]=inf;
d[start]=0;
heap.insert(make_pair(0,start));
while(!heap.empty())
{
c=heap.begin()->first;
nod=heap.begin()->second;
heap.erase(heap.begin());
for(it=a[nod].begin();it!=a[nod].end();it++)
{
next=it->first;
c=it->second;
if(d[next]>d[nod]+c)
{
if(!d[next])
heap.erase(heap.find(make_pair(d[next],next)));
d[next]=d[nod]+c;
heap.insert(make_pair(d[next],next));
}
}
}
}
bool verif()
{
int i;
for(i=1;i<=n;i++)
if(d[i]!=cost[i])
return false;
return true;
}
int main()
{
int i,j;
fin>>t;
for(i=1;i<=t;i++)
{
citire();
dijkstra();
if(verif()==true)
fout<<"DA";
else
fout<<"NU";
fout<<'\n';
for(j=1;j<=n;j++)
a[j].clear();
}
return 0;
}