Cod sursa(job #566726)

Utilizator catalin93Catalin Ionescu catalin93 Data 29 martie 2011 11:24:28
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.55 kb
#include<stdio.h>
#include<queue>
#include<vector>
#include<algorithm>
#include<string.h>

using namespace std;
const int N=50715;
const int INF=1000000000;

struct succesor 
{
	int nod,cost;
};

vector<succesor> a[N];
priority_queue<pair<int,int> > h;
int n,d[N],x[N],t;
bool s[N];

void citire()
{
	int m,i,x,y,c;
	scanf("%d%d",&n,&m);
	for(i=1;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&c);
		a[x].push_back((succesor){y,c});
	}
}
	
void actualizare(int x)
{
	int c,y;
	for(size_t i=0 ; i<a[x].size() ; i++)
	{
		y = a[x][i].nod;
		c = a[x][i].cost;
		if(d[x] + c < d[y])
		{
			d[y] = d[x] + c;
			h.push(make_pair(-d[y],y));
		}
	}
}


void init()
{
	for(int i=0 ; i<=n ; i++)
		d[i] = INF;
}

int vf_cost_min()
{
	while(!h.empty())
	{
		if(!s[h.top().second])
			return h.top().second;
		h.pop();
	}
	return 0;
}

void dijkstra(int x0)
{
	int x;
	d[x0] = 0;
	h.push(make_pair(d[x0],x0));
	for(int i=1 ; i<n ; i++)
	{
		x = vf_cost_min();
		if(d[x] == INF)
			return;
		s[x] = true;
		actualizare(x);
	}
}
int verificare()
{
	int i;
	for(i=1;i<=n;i++)
	{
		if(d[i] != x[i])
			return 1;
	}
	return 0;
}
int main()
{
	freopen("distante.in","r",stdin);
	freopen("distante.out","w",stdout);
	
	scanf("%d",&t);
	
	while(t--)
	{
		scanf("%d%d%d",&n,&m,&x0);
		for(i=1;i<=n;i++)
			scanf("%d",&x[i]);
		a.swap(vector<int>());
		memset(d,0,sizeof(d));
		memset(s,0,sizeof(s));
		
		citire();
		dijkstra(x0);
		if(verificare()==1) printf("NU");
		else
			printf("DA");
	}
	return 0;
}