Cod sursa(job #2094697)

Utilizator wilson182Alexandrina Panfil wilson182 Data 26 decembrie 2017 13:39:37
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.89 kb
#include<bits/stdc++.h>
using namespace std;
const int N=50020, INF=999999999;
int main(){
	ifstream f("distante.in");
	ofstream g("distante.out");
	int t;
	f>>t;
	while(t--){
		vector <pair<int,int> > mda[N];
		int d[N], d1[N];
		int n, m, s;
		f>>n>>m>>s;
		for(int i=1;i<=n;i++){
			d[i]=INF;
			f>>d1[i];
		}
		while(m--){
			int x, y, z;
			f>>x>>y>>z;
			mda[x].push_back({y,z});
		}
		d[s]=0;
		set <pair<int,int> > h;
		h.insert({0,s});
		while(!h.empty()){
			int nod=h.begin()->second;
			int l=mda[nod].size();
			h.erase(h.begin());
			for(int i=0;i<l;i++){
				int y=mda[nod][i].first, cost=mda[nod][i].second;
				if(d[y]>d[nod]+cost){
					if(d[y]!=INF){
						h.erase(h.find({d[y],y}));
					}
					d[y]=d[nod]+cost;
					h.insert({d[y],y});
				}
			}
		}
		int i;
		for(i=1;i<=n && d[i]==d1[i];i++);
		if(i==n+1)g<<"DA\n"; else g<<"NU\n";
	}
}