Cod sursa(job #554910)

Utilizator Cristy94Buleandra Cristian Cristy94 Data 15 martie 2011 10:29:06
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.27 kb
#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
#define Nmax 50010
#define dest first
#define cost second
using namespace std;

int c[Nmax],min_c[Nmax];
bool viz[Nmax],ok=1;

struct cmp{
	bool operator() (const int&a,const int&b)const{
		return (min_c[a]<min_c[b]);
	}
};

vector <pair <int,int> > a[Nmax];
priority_queue <int,vector <int>,cmp> q;



void dijkstra_check(int nod,int cost){
	viz[nod]=1;
   	for(unsigned int i=0;i<a[nod].size()&&ok;++i){
		if(!viz[a[nod][i].dest]){
			min_c[a[nod][i].dest]=cost+a[nod][i].cost;
			if(min_c[a[nod][i].dest] != c[a[nod][i].dest] ){
				ok=0;return;
			}
			q.push(a[nod][i].dest);
			
		}
			
	}
}
int main(){
	freopen("distante.in","r",stdin);
	freopen("distante.out","w",stdout);
	int T;
	for(scanf("%d",&T); T; --T){
		int N,M,S;
		scanf("%d%d%d",&N,&M,&S);
		
		memset(viz,0,sizeof(viz));
		memset(min_c,0,sizeof(min_c));
		ok=1;
		for(int i=1;i<=N;++i){
			scanf("%d",&c[i]);
			a[i].clear();
		}
		
		for(int i=1;i<=M;++i){
			int x,y,c;
			scanf("%d%d%d",&x,&y,&c);
			a[x].push_back(make_pair(y,c));
			a[y].push_back(make_pair(x,c));
	     }
   
		q.push(S);
		    dijkstra_check(S,0);
			if(ok)
				printf("DA\n");
			else
				printf("NU\n");
	}
return 0;
}