Cod sursa(job #1583312)

Utilizator robertkarolRobert Szarvas robertkarol Data 28 ianuarie 2016 21:07:24
Problema Ubuntzei Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.59 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],d[2001];
void dij(int p)
{

    int MIN,i,k,poz;
    for(k=1;k<=n-1;k++)
    {
        //cout<<p<<" ";
        s[p]=1; MIN=inf;
        for(i=1;i<=n;i++)
            if(d[p]+a[p][i]<d[i])
              d[i]=d[p]+a[p][i];
        for(i=1;i<=n;i++)
            if(d[i]<MIN&&!s[i])
            {
                MIN=d[i];
                poz=i;
            }
      //cout<<MIN<<" ";
      p=poz;
    }
}
int main()
{
    fin>>n>>m;
    fin>>p;
    for(i=1;i<=p;i++)
       {
           fin>>v[i];
           s[v[i]]=1;
           s[0]++;
       }
    v[0]=1;
    for(i=1;i<=n;i++)
       {
           for(j=1;j<=n;j++)
          if(i!=j) a[i][j]=inf;
          d[i]=inf;
          s[i]=!s[i];
       }
    //for(i=1;i<=n;i++) cout<<s[i]<<" ";
    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++)
        d[i]=a[1][i];
//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;
    dij(1);

    fout<<a[1][n];
    return 0;
}