Pagini recente » Cod sursa (job #2459515) | Cod sursa (job #1241533) | Cod sursa (job #12258) | Cod sursa (job #592001) | Cod sursa (job #1901035)
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
int t,n,m,s,x,a,b,i,ok,u,v[50005],r[50005],d[50005];
vector <int> w[11][50005];
vector <int> c[11][50005];
queue <int> q;
int main()
{
ifstream f("distante.in");
ofstream g("distante.out");
f>>t;
u=0;
while(u<t)
{
u++;
f>>n>>m>>s;
for(i=1; i<=n; i++)
{
f>>v[i];
d[i]=1;
w[u][i].push_back(0);
c[u][i].push_back(0);
r[i]=100000005;
}
for(i=1; i<=m; i++)
{
f>>a>>b>>x;
w[u][a][0]++;
w[u][a].push_back(b);
c[u][a].push_back(x);
w[u][b][0]++;
w[u][b].push_back(a);
c[u][b].push_back(x);
}
q.push(s);
d[s]=0;
r[s]=0;
ok=1;
while(!q.empty())
{
x=q.front();
q.pop();
if(d[x]==0)
{
d[x]=1;
for(i=1; i<=w[u][x][0]; i++)
{
if(r[w[u][x][i]]>r[x]+c[u][x][i])
{
r[w[u][x][i]]=r[x]+c[u][x][i];
d[w[u][x][i]]=0;
q.push(w[u][x][i]);
}
}
}
}
for(i=1; i<=n&&ok; i++) if(r[i]!=v[i]) ok=0;
if(ok) g<<"DA\n";
else g<<"NU\n";
}
f.close(); g.close();
return 0;
}