Pagini recente » Cod sursa (job #2216611) | Cod sursa (job #1389995) | Cod sursa (job #615413) | Cod sursa (job #1959810) | Cod sursa (job #1912660)
#include <fstream>
#include <queue>
#include <vector>
#include <bitset>
#define inf 1<<30
#define nmax 100005
using namespace std;
ifstream f("distante.in");
ofstream g("distante.out");
int n, m, d[nmax],d1[nmax],start;
vector<pair<unsigned short int,unsigned short int> >v[nmax];
vector<pair<unsigned short int,unsigned short int> >::iterator it;
bitset<nmax> uz;
int ok;
class comp{
public:
bool operator()(pair<unsigned short int,unsigned short int> a,pair<unsigned short int,unsigned short int> b)
{
return a.first>b.first;
}
};
priority_queue<pair<int,int>,vector<pair<int,int> >,comp >heap;
void dijsktra(int nod)
{ int x, i, y, z;
for(i=1;i<=n;i++)
d[i]=inf;
d[nod]=0;
heap.push(make_pair(0,nod));
while(!heap.empty())
{
x=heap.top().second;
heap.pop();
if(uz[x]==1)
continue;
uz[x]=1;
for(it=v[x].begin();it!=v[x].end();it++){
y=it->first;
z=it->second;
if(d[y]>d[x]+z){
d[y]=d[x]+z;
heap.push(make_pair(d[y],y));
}
}
}
d[nod]=0;
for(i=1;i<=n;i++)
if(d[i]!=d1[i])
{ g<<"NU"<<'\n';
ok=1;
break;
}
if(ok==0)
g<<"DA"<<'\n';
}
int main()
{ int i, x, y, c,t, sursa, k, j;
f>>t;
for(i=1;i<=t;i++)
{ f>>n>>m>>sursa;
for(j=1;j<=n;j++)
f>>d1[j];
for(k=1;k<=m;k++)
{ f>>x>>y>>c;
v[x].push_back(make_pair(y,c));
v[y].push_back(make_pair(x,c));
}
dijsktra(sursa);
}
}