Cod sursa(job #1712862)

Utilizator Bodo171Bogdan Pop Bodo171 Data 3 iunie 2016 21:30:25
Problema Distante Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <iostream>
#include<vector>
#include<fstream>
#include<set>
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);}

}
set< 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.insert(make_pair(0,source));
    while(!s.empty())
    {
        costul=(*s.begin()).first;
        x=(*s.begin()).second;
        s.erase(*s.begin());
        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.insert(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;
}