Pagini recente » Cod sursa (job #2777884) | Cod sursa (job #1575593) | Cod sursa (job #1054169) | Cod sursa (job #926426) | Cod sursa (job #857476)
Cod sursa(job #857476)
#include<fstream>
#include<vector>
#include<queue>
#include<cstdlib>
#define NMAX 50010
#define INF 100000010
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
struct muchie
{
int nod, cost;
};
int T, n, m, S, d[NMAX], sol[NMAX];
vector<muchie> a[NMAX];
queue<int> q;
void Citeste()
{
int i, x, y;
muchie z;
char c[40];
f>>c; n=atoi(c);
f>>c; m=atoi(c);
f>>c; S=atoi(c);
for (i=1; i<=n; ++i)
{
f>>c; sol[i]=atoi(c);
a[i].clear();
d[i]=INF;
}
for (i=1; i<=m; ++i)
{
f>>c; x=atoi(c);
f>>c; y=atoi(c);
f>>c; z.cost=atoi(c);
z.nod=y; a[x].push_back(z);
z.nod=x; a[y].push_back(z);
}
}
void Solve()
{
int i, x, lg;
muchie z;
d[S]=0; q.push(S);
while (!q.empty())
{
x=q.front(); q.pop();
for (i=0; i<a[x].size(); ++i)
{
z=a[x][i];
lg=z.cost+d[x];
if (lg==sol[z.nod])
{
d[z.nod]=lg;
q.push(z.nod);
}
}
}
}
void Scrie()
{
int i, ok=1;
for (i=1; i<=n; ++i)
if (sol[i]!=d[i])
{
ok=0;
break;
}
if (ok) g<<"DA\n"; else g<<"NU\n";
}
int main()
{
f>>T;
while (T--)
{
// Initializeaza();
Citeste();
Solve();
Scrie();
}
f.close();
g.close();
return 0;
}