Pagini recente » Cod sursa (job #573032) | Cod sursa (job #1475351) | Cod sursa (job #2903684) | Cod sursa (job #2037474) | Cod sursa (job #2120848)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#include <climits>
#define MAX 50001
#define INF INT_MAX
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int k, n, m, sursa, dist[MAX], d[MAX];
vector < pair < int , int > > v[MAX];
priority_queue < pair < int , int > > heap;
bitset < MAX > viz;
void init()
{
for(int i=1; i<=n; i++)
d[i]=INF;
}
void init_viz()
{
for(int i=1; i<=n; i++)
viz[i]=0;
}
bool djk();
void read()
{
f>>k;
for(int i=1; i<=k; i++)
{
init();
init_viz();
f>>n>>m>>sursa;
for(int j=1; j<=n; j++)
f>>dist[j];
int x, y, lungime;
for(int j=1; j<=m; j++)
{
f>>x>>y>>lungime;
v[x].push_back(make_pair(y, lungime));
v[y].push_back(make_pair(x, lungime));
}
if(djk()==true)
g<<"DA"<<endl;
else g<<"NU"<<endl;
}
}
bool djk()
{
heap.push(make_pair(0, sursa));
d[sursa]=0;
if(dist[sursa]!=0)
return false;
while(!heap.empty())
{
int node = heap.top().second;
heap.pop();
if(viz[node]==0)
{
viz[node]=1;
for(size_t i=0; i<v[node].size(); i++)
{
int p = v[node][i].second;
int q = v[node][i].first;
if(d[p]>d[node]+q)
{
d[p]=d[node]+q;
if(dist[p]>d[p])
return false;
heap.push(make_pair(-d[p], p));
}
}
}
for(int i=1; i<=n; i++)
if(dist[node]!=d[node] || d[node]==INF && dist[node]!=0)
return false;
}
}
int main()
{
read();
f.close();
g.close();
return 0;
}