Pagini recente » Cod sursa (job #2440049) | Cod sursa (job #2078912) | Cod sursa (job #360582) | Cod sursa (job #1571611) | Cod sursa (job #2654737)
#include <fstream>
#include <queue>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
vector<pair <int, int> > v[50003];
priority_queue< pair<int, int> , vector<pair<int, int> >, greater<pair<int, int> > > q;
int n, m, i, a, b, c, cost, el , d[50003],s , t[50003],x,aux[50003];
void sterge()
{
for(int i=1;i<=n;i++) v[i].clear();
}
int main()
{
fin>>x;
while(x>0)
{
fin>>n>>m>>s;
for(i=1;i<=n;i++)
{
fin>>aux[i];
d[i]=10000000;
t[i]=0;
}
d[s]=0;
for(i=1;i<=m;i++)
{
fin>>a>>b>>c;
v[a].push_back(make_pair(b, c));
v[b].push_back(make_pair(a, c));
}
for(i=0;i<v[s].size();i++)
{
q.push(make_pair(v[s][i].second, v[s][i].first));
t[v[s][i].first]=s;
}
int nr=n-1;
while(nr>0)
{
cost=q.top().first;
el=q.top().second;
q.pop();
if(d[el]>cost)
{
for(auto it:v[el]) if(t[el]!=it.first)
{
q.push(make_pair(it.second+cost, it.first));
t[it.first]=el;
}nr--;
d[el]=cost;
}
}
int ok=1;
for(i=1;i<=n;i++)
{
if(aux[i]!=d[i]) ok=0;
v[i].clear();
}
q.clear();
if(ok==0)
fout<<"NU\n";
else
fout<<"DA\n";
x--;
}
return 0;
}