Cod sursa(job #2759733)

Utilizator alex.prohnitchiAlex Prohnitchi alex.prohnitchi Data 20 iunie 2021 00:37:57
Problema Distante Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.91 kb
// by Alex Prohnitchi
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;

const ll mod=1e9+7;
const ll INF=2e9;
const int inff=INT_MAX;
const int dx[] = {0, 1, 0, -1};
const int dy[] = {1, 0, -1, 0};
const int N=100002;

#define all(x) x.begin(),x.end()
#define allr(x) x.rbegin(),x.rend()
#define rc(x)  return cout<<x<<"\n",0
#define sz(s)  (int) s.size()
#define pb push_back
#define mp make_pair
#define fr first
#define sc second
#define PI 3.14159265358979

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int rand(int l,int r){return uniform_int_distribution<int>(l,r)(rng);}

ifstream fin("distante.in");
ofstream fout("distante.out");

ll t,n;

vector<pii>v[N];
int rasp[N],cost[N],viz[N],d[N];

void solve() {
	
	
}

int main() {
ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	t=1;
	fin >> t;
	while (t--) {
			int n,m,s;
			fin >> n >> m >> s;
			int a,b,c;
			for (int i=1; i<=n; i++) {
				fin >> d[i];
			}
			for (int i=0; i<=n; i++) {
				cost[i]=inff;
				viz[i]=0;
				v[i].clear();
			}
			for (int i=1; i<=m; i++) {
				fin >> a >> b >> c;
				v[a].pb({b,c});
				v[b].pb({a,c});
			}
			cost[s]=0;
			priority_queue<pii>q;
			q.push({0,s});
			
			while(!q.empty()) {
				int nod=q.top().sc;
				q.pop();
				if (viz[nod])continue;
				viz[nod]=1;
				//cout << nod << '\n';
				for (auto it:v[nod]) {
					int cst=cost[nod]+it.sc;
					//cout << cost[nod]+cs << " " << cost[next] << '\n';
					if (cst<cost[it.fr]) {
						cost[it.fr]=cst;	
						q.push({-cost[it.fr],it.fr});	
					}
				}
				
			}
			
			bool ok=true;
			for (int i=1; i<=n; i++) {
				if (d[i]!=cost[i]) {
					ok=false;
					break;
				}
			}
			if (ok) fout << "DA\n";
			else fout << "NU\n";
	}

}