Cod sursa(job #1864446)

Utilizator jason2013Andronache Riccardo jason2013 Data 31 ianuarie 2017 19:28:15
Problema Ubuntzei Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.86 kb
#include<queue>
#include<vector>
#include<fstream>
using namespace std;
ifstream in("ubuntzei.in");
ofstream out("ubuntzei.out");
const int N=2010;
const int INF=0x3f3f3f;

int n,m,k,v[N],d[N],t[N],cost[18][18];
vector<int>a[N];
vector<int>c[N];


bool viz[N];
queue<int>coada;
int cc[131072][18];


inline int min(int eee,int www)
{
    return eee<www?eee:www;
}


void read()
{
    int x,y,z;

    in>>n>>m>>k;
    for(int i=1;i<=k;i++)
        in>>v[i];

    for(int i=1;i<=m;i++)
    {
        in>>x>>y>>z;
        a[x].push_back(y);
        a[y].push_back(x);
        c[x].push_back(z);
        c[y].push_back(z);
    }
}

void init()
{
    for(int i=0;i<=n;i++)
        d[i]=INF,viz[i]=0,t[i]=0;
}

void bford(int param)
{
    init();
    int r,x,costt;

    coada.push(param);
    d[param]=0;
    viz[param]=1;
    bool ok=1;

    while(ok && !coada.empty())
    {
        r=coada.front();
        coada.pop();
        viz[r]=0;

        for(size_t i=0;i<a[r].size();i++)
        {
            x=a[r][i];
            costt=c[r][i];
            if(d[r]+costt<d[x])
            {
                d[x]=d[r]+costt;
                if(!viz[x])
                {
                    viz[x]=true;
                    t[x]++;
                    if(t[x]>n-1)
                        ok=0;
                    coada.push(x);
                }
            }
        }
    }
}

void solve()
{
    int o=k+2,x;
    for(int i=0;i<=o;i++)
        for(int j=0;j<=o;j++)
            if(i!=j)
                cost[i][j]=INF;

    if(k==0)
    {
        bford(1);
        out<<d[n];
        return;
    }
    for(int i=1;i<=k;i++)
    {
        x=v[i];
        bford(x);

        cost[i][0]=cost[0][i]=d[1];
        cost[i][o-1]=cost[o-1][i]=d[n];

        for(int j=1;j<=k;j++)
            cost[i][j]=cost[j][i]=d[ v[j] ];
    }


    cost[0][0]=cost[o][o]=0;

    /*
    for(int i=0;i<o;i++)
    {
        for(int j=0;j<o;j++)
            out<<cost[i][j]<<"\t";
        out<<"\n";
    }
    */
    for(int i=0;i<(1<<o);i++)
        for(int j=0;j<o;j++)
            cc[i][j]=INF;

    cc[1][0]=0;

    for(int i=1;i<(1<<o);i++)
        for(int j=0;j<o;j++)
                if(i &(1<<j) )
                {
                    for(int k=0;k<=o;k++)
                    {
                        if(!cost[k][j] || cost[k][j]==INF)
                            continue;
                        if(i & (1<<k) )
                            cc[i][j]=min(cc[i][j], cc[i ^ (1<<j) ][k] + cost[k][j] );
                    }
                }
    /*
    for(int i=0; i < (1<<o) ; i++)
    {
        out<<i<<":\t";
        for(int j=0;j<=o;j++)
            out<<cc[i][j]<<"\t";
        out<<"\n";
    }
    */
    out<<cc[(1<<o)-1][o-1];
}

int main()
{
    read();
    solve();
    return 0;
}