Cod sursa(job #877294)

Utilizator SpiriFlaviuBerbecariu Flaviu SpiriFlaviu Data 12 februarie 2013 19:01:52
Problema Ubuntzei Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.77 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <queue>
#define INF 999999999

using namespace std;

ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
vector<pair<int,int> > v[2001];
int drum[17][2001];
queue<int> q;
int n,m,k,C[16],solmin=INF;
int sol[1<<16][16];
int used[16],lastused;

int get_min(int x,int next)
{
    if(x==0)
        return drum[0][next];
    int minx = INF;
    for(int dr=0;dr<k;dr++)
        if((x & (1<<dr)) == (1<<dr))
            if(minx > sol[x][dr] + drum[dr+1][next])
                minx = sol[x][dr] + drum[dr+1][next];
    return minx;
}

int main()
{

    fin>>n>>m>>k;
    for(int i=1;i<=k;i++)
        fin>>C[i];
    int x,co,y;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>co;
        v[x].push_back(make_pair(y,co));
        v[y].push_back(make_pair(x,co));
    }
    for(int i=0;i<=k;i++)
        for(int j=1;j<=n;j++)
            drum[i][j] = INF;
    C[0] = 1;
    for(int p = 0; p <= k; p++)
    {
        q.push(C[p]);
        drum[p][C[p]] = 0;
        while(!q.empty())
        {
            int nod = q.front();
            for(unsigned int i=0;i<v[nod].size();i++)
                if(drum[p][nod] + v[nod][i].second < drum[p][v[nod][i].first])
                {
                    q.push(v[nod][i].first);
                    drum[p][v[nod][i].first] = drum[p][nod] + v[nod][i].second;
                }
            q.pop();
        }
    }

    for(int nr=1;nr<(1<<k);nr++)
    {
        for(int dr = 0; dr<k;dr++)
            if((nr & (1<<dr)) == (1<<dr)) // ii vizitat
            {
                sol[nr][dr] = get_min(nr - (1<<dr),C[dr+1]);
            }
     }
     fout<<get_min((1<<k)-1,n);
     fin.close();
     fout.close();
     return 0;
}