Cod sursa(job #1304993)

Utilizator refugiatBoni Daniel Stefan refugiat Data 29 decembrie 2014 14:24:33
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.18 kb
#include<stdio.h>
#include<iostream>
#include<fstream>
#include<vector>
#include<queue>
#include<limits.h>
using namespace std;
#define nmax 50000
struct drum
{
    int d,l;
};
vector<drum> v[nmax];
queue<int> q;
int f[nmax];
void dij()
{
    int i,n,x;
    while(q.size()>0)
    {
        x=q.front();
        q.pop();
        n=v[x].size();
        for(i=0;i<n;++i)
        {
            if(f[v[x][i].d]>f[x]+v[x][i].l)
            {
                f[v[x][i].d]=f[x]+v[x][i].l;
                q.push(v[x][i].d);
            }
        }
    }
}
int main()
{
    ifstream si;
    si.open("distante.in");
    FILE* so=fopen("distante.out","w");
int t;
si>>t;
while(t--)
{
    int n,m,s;
    si>>n>>m>>s;
    --s;
    int x[n],i;
    for(i=0;i<n;++i)
        si>>x[i];
    int a,b,c;
    for(i=0;i<m;++i)
    {
        si>>a>>b>>c;
        --a;
        --b;
        v[a].push_back({b,c});
    }
    for(i=0;i<n;++i)
        f[i]=INT_MAX;
    f[s]=0;
    q.push(s);
    dij();
    for(i=0;i<n;++i)
        if(x[i]!=f[i])
            break;
    if(i<n)
    {
        fprintf(so,"NU\n");
    }
    else
        fprintf(so,"DA\n");
}
}