Cod sursa(job #929597)

Utilizator vandrei95Zamfir Vlad vandrei95 Data 27 martie 2013 09:46:42
Problema Ubuntzei Scor 85
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.91 kb
#include <iostream>
#include<fstream>
#define OO 100000000
#define N 2003
using namespace std;
int dc[17][N],cost[N][N],dist1[17][32800];

void Dijkstra(int x0,int n, int d[])
{
    int i,minim,vf;
    char viz[N]={0};
    for(i=1;i<=n;i++)
        d[i]=cost[x0][i];
    viz[x0]=1;
    do
    {
        minim=OO;
        for(i=1;i<=n;i++)
            if(minim>d[i]&&viz[i]==0)
            {
                minim=d[i];
                vf=i;
            }
        if(minim<OO)
        {
            viz[vf]=1;
            for(i=1;i<=n;i++)
                if(viz[i]==0&&d[vf]+cost[vf][i]<d[i])
                    d[i]=d[vf]+cost[vf][i];
        }
    }
    while(minim!=OO);
}
int main()
{
    fstream f("ubuntzei.in",ios::in),g("ubuntzei.out",ios::out);
    int x,y,i,j,n,m,k,c[17],d1[N],cst,nrSub,s,Min;
    f>>n>>m>>k;
    for(i=0;i<k;i++)
        f>>c[i];
     for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            if(i!=j)
                cost[i][j]=OO;
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>cst;
        cost[x][y]=cost[y][x]=cst;
    }
    Dijkstra(1,n,d1);
    if(k==0)
    {
        g<<d1[n];
        g.close();
        return 0;
    }
    for(i=0;i<k;i++)
        Dijkstra(c[i],n,dc[i]);
    nrSub=1<<k;
    for(s=1;s<nrSub;s++)
    {
        for(i=0;i<k;i++)
            if(s==(1<<i))//ci apartine S1|S|=1
            {
                dist1[i][s]=d1[c[i]];
                break;
            }
        if(i<k)
            continue;
        for(i=0;i<k;i++)
        {
            dist1[i][s]=OO;
            if(s&(1<<i))//c[i]apartine s
                for(j=0;j<k;j++)
                    if(s&(1<<j)&&j!=i)//c[j]apartine s
                        dist1[i][s]=min(dist1[i][s],dist1[j][s-(1<<i)]+dc[j][c[i]]);
        }
    }
    Min=OO;
    for(i=0;i<k;i++)
        Min=min(Min,dist1[i][nrSub-1]+dc[i][n]);
    g<<Min;
    f.close();
    g.close();
}