Cod sursa(job #1279486)

Utilizator Arodoet96Teodora Stoleru Arodoet96 Data 30 noiembrie 2014 14:18:45
Problema Ubuntzei Scor 45
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.82 kb
#include <cstdio>
#include <vector>
#define NMAX 2004
#define INF 1000000

using namespace std;

FILE*fin=fopen("ubuntzei.in", "r");
FILE*fout=fopen("ubuntzei.out", "w");

int n, m, k;
int oras[16];
struct vecin
{
    int varf;
    int cost;
};
vector <vecin> A[NMAX];
int cmin[16][NMAX];
bool Z[16][NMAX];
int p[16];
bool uz[16];
int minim;
long long int cost;

void citire();
void generare();
void initializare(int rang, int x);
void dijkstra(int rang);
void perm(int);
void solutie();

int main()
{
    citire();
    generare();
    minim=15*INF;
    if(!k)
    {
        fprintf(fout, "%d\n", cmin[0][n]);
        return 0;
    }
    perm(1);
    fprintf(fout, "%d\n", minim);
    return 0;
}

void citire()
{
    int i, x, y, c;
    vecin v;

    fscanf(fin, "%d %d %d\n", &n, &m, &k);//fin>>n>>m>>k;

    for(i=1;i<=k;i++)
        fscanf(fin, "%d ", &oras[i]);

    for(i=1;i<=m;i++)
    {
        fscanf(fin, "%d %d %d\n", &x, &y, &c);//fin>>x>>y>>c;

        v.varf=y; v.cost=c;
        A[x].push_back(v);

        v.varf=x;
        A[y].push_back(v);
    }
}

void generare()
{
    int i;

    initializare(0, 1);
    dijkstra(0);

    for(i=1;i<=k;i++)
    {
        initializare(i, oras[i]);
        dijkstra(i);
    }
}

void initializare(int rang, int x)
{
    int i, lg;

    for(i=1;i<=n;i++)
        cmin[rang][i]=INF;

    cmin[rang][x]=0;
    Z[rang][x]=1;

    lg=A[x].size();
    for(i=0;i<lg;i++)
        cmin[rang][A[x][i].varf]=A[x][i].cost;
}

void dijkstra(int rang)
{
    int i, j, lg;
    int minim;

    cmin[rang][0]=INF;

    for(i=1;i<=n;i++)
    {
        minim=0;
        for(j=1;j<=n;j++)
            if(!Z[rang][j] && cmin[rang][j]<cmin[rang][minim])
                minim=j;

        if(!minim)
            return;

        Z[rang][minim]=1;

        lg=A[minim].size();
        for(j=0;j<lg;j++)
            if(!Z[rang][A[minim][j].varf] && cmin[rang][A[minim][j].varf]>cmin[rang][minim]+A[minim][j].cost)
                cmin[rang][A[minim][j].varf]=cmin[rang][minim]+A[minim][j].cost;
    }
}

void perm(int nr)
{
    int i;
    if(nr==k+1)
    {
        //solutie();
        cost+=cmin[p[k]][n];
        if(cost<minim)
            minim=cost;

        cost-=cmin[p[k]][n];
        return;
    }

    for(i=1;i<=k;i++)
        if(!uz[i] && cmin[p[nr-1]][oras[i]]!=INF)
        {
            uz[i]=1;
            p[nr]=i;
            cost+=cmin[p[nr-1]][oras[i]];

            perm(nr+1);

            cost-=cmin[p[nr-1]][oras[i]];
            uz[i]=0;
        }
}

void solutie()
{
    int i;
    long long int cost;

    cost=cmin[0][oras[p[1]]];
    for(i=2;i<=k;i++)
        cost+=cmin[p[i-1]][oras[p[i]]];

    cost+=cmin[p[k]][n];

    if(cost<minim)
        minim=cost;
}