Cod sursa(job #1611710)

Utilizator alexmisto342Turdean Alexandru alexmisto342 Data 24 februarie 2016 12:58:00
Problema Ubuntzei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.88 kb
#include <fstream>
#include <iostream>
#include <queue>
#include <algorithm>
#include <vector>
#include <map>
#include <cstring>
#include <string>
#define x first
#define y second
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
int i,j,n,m,a,b,k,x,lx,ly,c,mini,is[2005],ul[2005];
vector <pair<int,int> > v[2005],newnod;
char f[2005][4100];
cost int inf = 2000000000;
queue <int> q;
int dp[(1<<17)+10][20],cost[20][20],c[2005];
void bf(int nod)
{
    int i;
    for(i = 1; i <= n; i++)
            c[i] = inf;
    q.push(nod);
    c[nod] = 0;
    while(q.size())
    {
        int now = q.front();
        for(i = 0; i < v[now].size(); i++)
            if(c[v[now][i].x] > c[now] + v[now][i].y)
                c[v[now][i].x] = c[now] + v[now][i].y, q.push(v[now][i].x);
        q.pop();
    }
    for(i = 0; i < newnod.size(); i++)
        cost[nod][newnod[i]] = cost[newnod[i]][nod] = c[newnod[i]];
}
int main()
{
    fin >> n >> m >> k;
    for(i = 1; i <= k; i++)
        fin >> a,newnod.push_back(a);
    newnod.push_back(1);
    newnod.push_back(n);
    k+=2;
    for(i = 1; i <= m; i++)
    {
        fin >> a >> b >> c;
        v[a].push_back(make_pair(b,c));
        v[b].push_back(make_pair(a,c));
    }
    for(i = 0; i <= k; i++)
    for(j = 0; j <= k; j++)
        cost[i][j] = inf;
    for(i = 1; i < newnod.size(); i++)
        bfs(newnod[i]);
    n = k ;
    for(i = 0 ; i < (1 << n); i++)
        for(j = 0; j < n; j++)
            dp[i][j] = inf;
    dp[1][0] = inf;
    for(i = 1; i < 1<<n; i++)
        for(j = 0; j < n; j++)
            if(i&(1<<j))
                for(a = 0; a < n; a++)
                    dp[i][j] = min(dp[i][j], dp[i ^ (1<<j)][a] + cost[a][j]);
    int mini = inf;
    for(i = 0 ; i < n ; i++)
        mini = min(mini, dp[(1<<n) - 1][i]);
    fout<<mini;

    return 0;
}