Pagini recente » Cod sursa (job #1567134) | Cod sursa (job #526054) | Cod sursa (job #1545848) | Cod sursa (job #1282953) | Cod sursa (job #1771097)
#include <fstream>
#include <vector>
#include <queue>
#define VAL 50005
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
int T, N, M, i, j;
int a, b, c, S, k;
int cost[VAL], x[VAL];
vector<int> v[VAL];
vector<int> p[VAL];
queue<int> q;
bool ok;
void bfs()
{
q.push(S);
while (q.empty()==false)
{
a=q.front();
q.pop();
for (i=0; i<v[a].size(); i++)
{
if (cost[v[a][i]]==-1)
{
cost[v[a][i]]=cost[a]+p[a][i];
q.push(v[a][i]);
}
else
{
if (cost[v[a][i]]>cost[a]+p[a][i])
{
cost[v[a][i]]=cost[a]+p[a][i];
q.push(v[a][i]);
}
}
}
}
}
int main()
{
fin >> T;
for (k=1; k<=T; k++)
{
fin >> N >> M >> S;
ok=true;
for (i=1; i<=N; i++)
fin >> x[i];
for (i=1; i<=M; i++)
{
fin >> a >> b >> c;
v[a].push_back(b);
p[a].push_back(c);
}
if (x[S]!=0)
{
ok=false;
fout << "NU\n";
}
else
{
for (i=1; i<=N; i++)
if (i!=S)
cost[i]=-1;
bfs();
for (i=1; i<=N; i++)
{
if (cost[i]!=x[i])
ok=false;
cost[i]=0;
v[i].clear();
p[i].clear();
}
if (ok==true)
fout << "DA\n";
else
fout << "NU\n";
}
}
fin.close();
fout.close();
return 0;
}