Cod sursa(job #1912660)

Utilizator nartorrewrew narto Data 8 martie 2017 10:06:15
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.59 kb
#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);
    }
}