Pagini recente » Cod sursa (job #276663) | Cod sursa (job #1211600) | Cod sursa (job #364086) | Cod sursa (job #1814847) | Cod sursa (job #2375949)
#include <bits/stdc++.h>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
#define nmax 50002
const int inf=(1<<31)-1;
int d[nmax];
int ver[nmax];
bool viz[nmax];
void reinitializare(int n)
{
for(int i=1; i<=n; i++)
viz[i]=0;
}
struct cmp
{
bool operator()(int x,int y)
{
return d[x]>d[y];
};
};
vector < pair <int, int> > v[100001];
priority_queue<int, vector<int> , cmp>q;
int n,m,t,start,p;
void dijkstra(int sursa)
{
///toare distante sunt initial egale cu infinit
for(int i=1; i<=n; ++i)
d[i]=inf;
d[sursa]=0;
viz[sursa]=1;
q.push(sursa);
while(!q.empty())
{
int nod=q.top();
viz[nod]=0;
q.pop();
for(size_t i=0; i<v[nod].size(); ++i)
{
int vecin=v[nod][i].first;
int cost=v[nod][i].second;
if(d[vecin]>d[nod]+cost)
{
d[vecin]=d[nod]+cost;
if(!viz[vecin])
{
q.push(vecin);
viz[vecin]=1;
}
}
}
}
}
int main()
{
f>>p;
for(int k=1; k<=p; k++)
{
bool ok=1;
f>>n>>m>>start;
for(int i=1; i<=n; i++)
{
f>>ver[i];
v[i].clear();
}
for(int i=1; i<=m; ++i)
{
int x,y,c;
f>>x>>y>>c;
v[x].push_back(make_pair(y,c));
}
dijkstra(start);
for(int i=1; i<=n&&ok; i++)
if(d[i]!=ver[i])ok=0;
reinitializare(n);
if(ok==1)g<<"DA"<<'\n';
else g<<"NU"<<'\n';
}
f.close();
g.close();
return 0;
}