Pagini recente » Cod sursa (job #2058203) | Cod sursa (job #2667215) | Cod sursa (job #1886032) | Cod sursa (job #854245) | Cod sursa (job #2797018)
#include <queue>
#include <fstream>
using namespace std;
#define N_MAX 50000
#define INF (1 << 30)
priority_queue<pair<int, int>> dijkstra;
vector<pair<int, int>> G[N_MAX];
int ans[N_MAX], expected_ans[N_MAX];
void DIJKSTRA() {
while ( !dijkstra.empty() ) {
pair<int, int> elem = dijkstra.top();
dijkstra.pop();
if ( ans[elem.second] == -INF || ans[elem.second] == elem.first ) {
for ( auto copil : G[elem.second] ) {
if ( ans[copil.first] < elem.first + copil.second * -1 ) {
ans[copil.first] = elem.first + copil.second * -1;
dijkstra.push( {ans[copil.first], copil.first} );
}
}
}
}
}
ifstream cin ( "distante.in" );
ofstream cout ( "distante.out" );
int main() {
int t, i, j, n, m, s, a, b, cost;
cin >> t;
for ( i = 0; i < t; i++ ) {
cin >> n >> m >> s;
for ( j = 0; j <= n; j++ )
ans[j] = -INF;
ans[s] = -1;
for ( j = 1; j <= n; j++ ) {
cin >> expected_ans[j];
G[j].clear();
}
for ( j = 1; j <= m; j++ ) {
cin >> a >> b >> cost;
G[a].emplace_back(b, cost);
G[b].emplace_back(a, cost);
}
dijkstra.push({-1, s});
DIJKSTRA();
j = 1;
while ( j <= n && ( ans[j] + 1 ) * -1 == expected_ans[j] ) {
j++;
}
if ( j <= n )
cout << "NU\n";
else
cout << "DA\n";
}
return 0;
}