Cod sursa(job #1583278)

Utilizator robertkarolRobert Szarvas robertkarol Data 28 ianuarie 2016 20:33:29
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.35 kb
#include <iostream>
#include <fstream>
#define inf 2000000000
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
int n,m,k,i,j,p,v[2001],cost,x,y,c,s[2001],MIN;
long long a[2001][2001];
int suma(int nod, int nr, int sum, int &MIN)
{
    //cout<<nod<<" ";
    s[nod]=1;
    if(nr==p&&a[nod][n]>0&&a[nod][n]<inf)
    {
        cout<<1;
        if(sum+a[nod][n]<MIN||!MIN) MIN=sum+a[nod][n];
        s[nod]=0;
    }
    else
    {
        int i;
        for(i=1;i<=p;i++)
            if(!s[v[i]]&&a[p][v[i]]>0&&a[p][v[i]]<inf)
                suma(v[i],nr+1,sum+a[p][v[i]],MIN);
        s[nod]=0;
    }
}
int main()
{
    fin>>n>>m;
    fin>>p;
    for(i=1;i<=p;i++)
        fin>>v[i];
    v[0]=1;
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
          if(i!=j) a[i][j]=inf;
    for(i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        a[x][y]=a[y][x]=c;
    }

    for(k=1;k<=n;k++)
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
              if(a[i][j]>a[i][k]+a[k][j]&&i!=k&&k!=j)
                a[i][j]=a[i][k]+a[k][j];
//for(i=1;i<=n;i++)
//    {
//        for(j=1;j<=n;j++)
//            cout<<a[i][j]<<" ";
//        cout<<"\n";
//    }
//    for(i=1;i<=p;i++)
//       cost+=a[v[i-1]][v[i]];
//    fout<<cost+a[v[p]][n];
    suma(1,0,0,MIN);
    fout<<MIN;
    return 0;
}