Pagini recente » Cod sursa (job #2102965) | Cod sursa (job #367336) | Cod sursa (job #1529668) | Cod sursa (job #54204) | Cod sursa (job #650701)
Cod sursa(job #650701)
#include<fstream>
#include<vector>
#include<queue>
using namespace std;
int T,n,m,x0,d[50010],v[50010];
struct Muchie{int nod,cost;};
vector <Muchie> G[50010];
void Bellman_Ford()
{
int i;
for(i=1;i<=n;i++)
d[i]=2000000000;
queue <int> Q;
int x,nod,c;
Q.push(x0);
d[x0]=0;
while(!Q.empty())
{
x=Q.front();
Q.pop();
for(i=0;i<G[x].size();i++)
{
nod=G[x][i].nod;
c=G[x][i].cost;
if(d[nod]>d[x]+c)
{
d[nod]=d[x]+c;
Q.push(nod);
}
}
}
}
bool Verificare()
{
int i;
for(i=1;i<=n;i++)
if(d[i]!=v[i])
return false;
return true;
}
int main()
{
int t,i,x,y,c;
Muchie aux;
bool ok;
ifstream fin("distante.in");
ofstream fout("distante.out");
fin>>T;
for(t=0;t<T;t++)
{
fin>>n>>m>>x0;
for(i=1;i<=n;i++)
fin>>v[i];
for(i=1;i<=m;i++)
{
fin>>x>>y>>c;
aux.nod=y; aux.cost=c;
G[x].push_back(aux);
aux.nod=x;
G[y].push_back(aux);
}
Bellman_Ford();
ok=Verificare();
if(ok)
fout<<"DA\n";
else
fout<<"NU\n";
for(i=1;i<=n;i++)
G[i].clear();
}
fin.close();
fout.close();
return 0;
}