Pagini recente » Cod sursa (job #2916678) | Cod sursa (job #1317609) | Cod sursa (job #390994) | Cod sursa (job #3202537) | Cod sursa (job #1089983)
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define INF 0x3f3f3f3f
#define NMAX 50002
using namespace std;
ifstream fin ("distante.in"); ofstream fout ("distante.out");
int T, N, M, cost[NMAX], start, dZ[NMAX];
vector <int> v[NMAX];
queue <int> q;
void ReadData ()
{
fin >> N >> M >> start;
for (int i = 1; i <= N; ++i)
fin >> dZ[i];
for (int i = 1, a, b, c; i <= M; ++i)
{
fin >> a >> b >> c;
v[a].push_back (b);
v[a].push_back (c);
v[b].push_back (a);
v[b].push_back (c);
}
for (int i = 1; i <= N; ++i)
{
if (i != start)
cost[i] = INF;
}
}
void Solve ()
{
q.push (start);
while (!q.empty ())
{
int x = q.front ();
for (int i = 0; i < v[x].size (); i += 2)
{
if (cost[v[x][i]] > cost[x] + v[x][i + 1])
{
q.push (v[x][i]);
cost[v[x][i]] = cost[x] + v[x][i + 1];
}
}
q.pop ();
}
}
int Check ()
{
cost[start] = 0;
for (int i = 1; i <= N; ++i)
{
if (cost[i] != INF)
{
if (cost[i] != dZ[i]) return 0;
}
else return 0;
}
return 1;
}
int main ()
{
fin >> T;
for (int i = 1; i <= T; ++i)
{
ReadData ();
Solve ();
if (Check ())
fout << "DA\n";
else
fout << "NU\n";
}
fin.close (); fout.close ();
return 0;
}