Cod sursa(job #1712894)

Utilizator Bodo171Bogdan Pop Bodo171 Data 3 iunie 2016 23:06:06
Problema Distante Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.23 kb
#include <iostream>
#include<vector>
#include<fstream>
#include<set>
#include<queue>
using namespace std;
vector<int> v[50005],c[50005];
int cost[50005],costul,i,n,m,t,source,match[50005],x,y;
bool matching;
inline void reset()
{
    for(i=1;i<=n;i++)
        {v[i].clear();c[i].clear();cost[i]=(1<<30);}

}
queue< pair<int,int> >s;

int main()
{
    ifstream f("distante.in");
    ofstream g("distante.out");
    f>>t;
    for(int cnt=1;cnt<=t;cnt++)
    {
    f>>n>>m>>source;
    reset();
    cost[source]=0;matching=1;
    for(i=1;i<=n;i++) f>>match[i];
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>costul;
        v[x].push_back(y);
        v[y].push_back(x);
        c[x].push_back(costul);
        c[y].push_back(costul);
    }
    s.push(make_pair(0,source));
    while(!s.empty())
    {
        costul=s.front().first;
        x=s.front().second;
        s.pop();
        for(i=0;i<v[x].size();i++)
            if(c[x][i]+costul<cost[v[x][i]])
        {
            cost[v[x][i]]=c[x][i]+costul;
            s.push(make_pair(c[x][i]+costul,v[x][i]));
        }
    }
    for(i=1;i<=n;i++)
    {if(match[i]!=cost[i]) matching=0;}
    if(matching) g<<"DA\n";
    else g<<"NU\n";
    }
    return 0;
}