Pagini recente » Cod sursa (job #2531013) | Cod sursa (job #465654) | Cod sursa (job #2807471) | Cod sursa (job #2374588) | Cod sursa (job #921515)
Cod sursa(job #921515)
#include<cstdio>
#include<vector>
#include<algorithm>
#include<queue>
#include<bitset>
#define Nmax 50001
#define mp make_pair
using namespace std;
int Dinit[Nmax], D[Nmax], N, M, S, L;
vector<pair<int, int> > G[Nmax];
bitset<Nmax> viz;
void dijkstra() {
priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > PQ;
int x, cost, nod;
vector<pair<int, int> > :: iterator it, itend;
D[S] = 0;
PQ.push(mp(0, S));
while(!PQ.empty()) {
x = PQ.top().second;
if(viz[x]) {
PQ.pop();
continue;
}
D[x] = PQ.top().first;
viz[x] = 1;
PQ.pop();
for(it=G[x].begin(), itend=G[x].end(); it!=itend; ++it) {
nod = it->first;
cost = it->second;
if(D[nod] > D[x] + cost) {
D[nod] = D[x] + cost;
PQ.push(mp(D[nod], nod));
}
}
}
}
int verif() {
for(int i=1; i<=N; i++)
if(D[i] != Dinit[i])
return 0;
return 1;
}
int main() {
freopen("distante.in","r",stdin);
freopen("distante.out","w",stdout);
int T, i, j, c;
scanf("%d",&T);
while(T--) {
scanf("%d %d %d",&N,&M,&S);
for(i=1; i<=N; i++) {
scanf("%d",&Dinit[i]);
G[i].clear();
D[i] = 1<<30;
viz[i] = 0;
}
while(M--) {
scanf("%d %d %d",&i,&j,&c);
G[i].push_back(make_pair(j,c));
G[j].push_back(make_pair(i,c));
}
dijkstra();
if(verif() == 1)
printf("DA\n");
else
printf("NU\n");
}
return 0;
}