Cod sursa(job #1229137)

Utilizator TudorMTudor Moldovanu TudorM Data 16 septembrie 2014 16:45:01
Problema Ubuntzei Scor 25
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.59 kb
#include<fstream>
#include<iostream>
#include<queue>
#include<vector>
#define INF 100000000
using namespace std;
ifstream f("ubuntzei.in");
ofstream g("ubuntzei.out");
int n, m, a[201][201],prieteni[2001], l, total=INF, s, st[201];
queue <int> c;
typedef struct
{
    int y, cost;
}MUCHIE;
vector <MUCHIE> v[201];
MUCHIE w;
void citire()
{
    int i, j, x, y, pret;
    f>>n>>m>>l;
    for(i=1;i<=l;i++)f>>prieteni[i];
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>pret;
        a[x][y]=a[y][x]=pret;
    }
}
void rf()
{
    int k, i, j;
    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]||a[i][j]==0)&&a[i][k]&&a[k][j]&&i!=j)a[i][j]=a[i][k]+a[k][j];
            }
        }
    }
}
int valid(int p)
{
    int i, j;
    for(i=1;i<p;i++)if(st[i]==st[p])return 0;
    return 1;
}
void calcul(int p)
{
    int i, j;
    s=0;
    s+=a[1][st[1]];
    for(i=2;i<=p;i++)
    {
        s+=a[st[i-1]][st[i]];
    }
    s+=a[st[p]][n];
}
void bkt(int p)
{
    int i, j;
    for(i=1;i<=l;i++)
    {
        st[p]=prieteni[i];
        if(valid(p))
        {
            if(p==l)
            {
                calcul(p);
                if(s<total)total=s;
            }
            else bkt(p+1);
        }

    }
}
int main()
{
    int i, j, k;
    citire();
    rf();
    /*for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)cout<<a[i][j]<<" ";
        cout<<'\n';
    }*/
    bkt(1);
    g<<total;
    f.close();
    g.close();
    return 0;
}