Cod sursa(job #886025)

Utilizator ard_procesoareLupicu ard_procesoare Data 22 februarie 2013 16:57:15
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("distante.in");
ofstream fout("distante.out");
#define NMAX 50005
int n,m,drum[NMAX],sol[NMAX],T,X,pas;
vector <int> v[NMAX],cost[NMAX];
queue <int> qnod;
bool ever[NMAX],viz[NMAX];
void read()
{
    fin>>n>>m>>X;
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        fin>>a>>b>>c;
        v[a].push_back(b);
        cost[a].push_back(c);
    }
}
void tipar()
{
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<v[i].size();j++)
        {
            fout<<i<<" "<<v[i][j]<<" "<<cost[i][j]<<endl;
        }
    }
}
void bfs(int val)
{
    int nod;
    qnod.push(1);
    while(!qnod.empty())
    {
        nod=qnod.front();
        qnod.pop();
        ever[nod]=true;
        viz[nod]=false;
        for(int i=0;i<v[nod].size();i++)
        {
            if(drum[v[nod][i]] > drum[nod]+cost[nod][i] || ever[v[nod][i]]== false)
            {
                ever[v[nod][i]]=true;
                drum[v[nod][i]] = drum[nod]+cost[nod][i];
                if(viz[v[nod][i]] == false)
                    qnod.push(v[nod][i]);
                viz[v[nod][i]]=true;
            }
        }
    }
}
bool check()
{
    for(int i=1;i<=n;i++)
    {
        if(drum[i]!=sol[i]) return 0;
    }
    return 1;
}
void init()
{
    for(int i=1;i<=n;i++)
        drum[i]=ever[i]=viz[i]=0;
}
int main()
{
    fin>>T;
    for(pas=1;pas<=T;pas++)
    {
        read();
        init();
        bfs(X);
        if(check()) fout<<"DA\n";
        else fout<<"NU\n";
    }
}