Pagini recente » Cod sursa (job #2660554) | Cod sursa (job #1289710) | Cod sursa (job #1171335) | Cod sursa (job #1337336) | Cod sursa (job #2375825)
#include <bits/stdc++.h>
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
#define nmax 50002
///reprezinta o variabila foarte mare, cea mai mare ce o poate memora int ul
const int inf=(1<<31)-1;
int d[nmax];
int ver[nmax];
bool viz[nmax];
void reinitializare(int n)
{
for(int i=1; i<=n; i++)
viz[i]=0;
}
struct cmp
{
///se ocupa de punerea in coada de prioritate in ordine dupa distante
bool operator()(int x,int y)
{
return d[x]>d[y];
};
};
vector < pair <int, int> > v[100001];
priority_queue<int, vector<int> , cmp>q;
int n,m,t,start,p;
void dijkstra(int sursa)
{
///toare distante sunt initial egale cu infinit
for(int i=1; i<=n; ++i)
d[i]=inf;
///reinitializam distanta de plecare cu 0, intrucat distanta de la un nod la el insusi este 0
d[sursa]=0;
viz[sursa]=1;
///punem in coada de prioritati nodul si incepem dijkstra
q.push(sursa);
while(!q.empty())
{
///luam nodul cu lungimea cea mai mica si il eliminam din coada
int nod=q.top();
viz[nod]=0;
q.pop();
///la nodul actual cautam toate legaturile si verificam daca distanta pana la vecin> decat distanta pana la nod + costul muchiei
///daca se intampla o schimbam
///Practic el actualizeaza distanta de la un nod pana la celelalte noduri verificand daca legatura ce o gaseste acum este mai mica decat cele gasite anterior
for(size_t i=0; i<v[nod].size(); ++i)
{
int vecin=v[nod][i].first;
int cost=v[nod][i].second;
if(d[vecin]>d[nod]+cost)
{
d[vecin]=d[nod]+cost;
if(!viz[vecin])
{
q.push(vecin);
viz[vecin]=1;
}
}
}
}
}
int main()
{
f>>p;
for(int k=1; k<=p; k++)
{
bool ok=1;
f>>n>>m>>start;
for(int i=1; i<=n; i++)
{
f>>ver[i];
v[i].clear();
}
for(int i=1; i<=m; ++i)
{
int x,y,c;
f>>x>>y>>c;
v[x].push_back(make_pair(y,c));
}
dijkstra(start);
for(int i=1; i<=n&&ok; i++)
if(d[i]!=ver[i])ok=0;
reinitializare(n);
if(ok)g<<"DA"<<'\n';
else g<<"NU"<<'\n';
}
f.close();
g.close();
return 0;
}