Pagini recente » Cod sursa (job #1873541) | Cod sursa (job #1044774) | Cod sursa (job #2781112) | Cod sursa (job #669951) | Cod sursa (job #2447322)
#include <bits/stdc++.h>
#define fi first
#define sc second
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int q, n, m, S, B[50001], x, y, z;
bool viz[50002];
int dis[50001];
vector<pair<int, int>>v[50001];
set<pair<int, int>>s;
void dfs(int nod)
{
viz[nod]=1;
for(int i=0; i<v[nod].size(); i++)
{
if(!viz[v[nod][i].fi])
{
viz[v[nod][i].fi]=1;
s.insert(make_pair(v[nod][i].sc+dis[nod], v[nod][i].fi));
dis[v[nod][i].fi]=v[nod][i].sc+dis[nod];
}
else
{
if(v[nod][i].sc+dis[nod]<dis[v[nod][i].fi])
{
s.erase(make_pair(dis[v[nod][i].fi], v[nod][i].fi));
dis[v[nod][i].fi]=v[nod][i].sc+dis[nod];
s.insert(make_pair(v[nod][i].sc+dis[nod], v[nod][i].fi));
}
}
}
}
int main()
{
f>>q;
for(; q; q--)
{
bool ok=1;
s.clear();
f>>n>>m>>S;
for(int i=1; i<=n; i++)
v[i].clear();
for(int i=1; i<=n; i++)
f>>B[i];
for(int i=1; i<=m; i++)
{
f>>x>>y>>z;
v[x].push_back(make_pair(y, z));
v[y].push_back(make_pair(x, z));
}
s.insert(make_pair(0, S));
while(!s.empty())
{
int a=(*s.begin()).sc;
s.erase(s.begin());
dfs(a);
}
for(int i=1; i<=n; i++)
if(dis[i]!=B[i])
{
i=n+1;
ok=0;
g<<"NU";
}
if(ok==1)
g<<"DA";
g<<'\n';
}
return 0;
}