Cod sursa(job #1304720)

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