Cod sursa(job #1316500)

Utilizator ade_tomiEnache Adelina ade_tomi Data 13 ianuarie 2015 21:10:52
Problema Distante Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.42 kb
#include<iostream>
#include<fstream>
#include<queue>
#include<vector>
#define inf 0x3f3f3f3f
using namespace std;
priority_queue<pair<int,int> , vector<pair<int, int> > , greater<pair<int,int> > > q;
vector<pair<int, int > > v[50004];
int i,n,d[50004],m,sol[50004],t,shp,s,ok,a,b,c,u,x;
int main()
{
    ifstream cin("distante.in");
    ofstream cout("distante.out");
    cin>>t;
    for(shp=1;shp<=t;shp++)
    {
        cin>>n>>m>>s;
        for(i=1;i<=n;i++)
        {
            cin>>sol[i];
            d[i]=inf;
            v[i].clear();
        }
        for(i=1;i<=m;i++)
        {
            cin>>a>>b>>c;
            v[a].push_back(make_pair(b,c));
            v[b].push_back(make_pair(a,c)); 
        }
        d[s]=0;
        q.push(make_pair(0,s));
        while(!q.empty())
        {
            c=q.top().first;
            x=q.top().second;
            q.pop();
            for(vector<pair<int,int> > :: iterator it=v[x].begin();it!=v[x].end();it++)
            {
                if(d[it->first]>c+it->second)
                {
                    d[it->first]=c+it->second;
                    q.push(make_pair(d[it->first],it->first));
                }

            }
        }
        ok=0;
        for(i=1;i<=n;i++)
        {
            if(d[i]==sol[i]||(d[i]==inf && sol[i]==0))
                continue;
            ok=1;
        }
        if(ok==0)
            cout<<"DA\n";
        else cout<<"NU\n";

    }
    return 0;
}