Cod sursa(job #1617771)

Utilizator bogdanboboc97Bogdan Boboc bogdanboboc97 Data 27 februarie 2016 16:15:55
Problema Ubuntzei Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.67 kb
#include <iostream>
#include <vector>
#include <algorithm>
#include <limits>
#include <numeric>
#include <cstring>
#include <string>
#include <queue>
#include <set>
#include <fstream>
#define pb push_back
#define mp make_pair
#define INF numeric_limits<int>::max()
#define bit(x) (-x)&x
#define int64 long long
using namespace std;
ifstream in("ubuntzei.in");
ofstream out("ubuntzei.out");
vector< vector< pair<int,int> > > a;
queue<int> q;
vector<bool> use;
int n,m,k,c[20];
void bellman(int st,vector<int>& d)
{
    use=vector<bool> (n+1);
    d=vector<int> (n+1,INF);
    q.push(st);
    use[st]=true;
    d[st]=0;
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        use[x]=false;
        for(vector< pair<int,int> >::iterator i=a[x].begin();i!=a[x].end();i++)
        if(d[x] + i->second < d[i->first])
        {
            d[i->first]=d[x] + i->second;
            if(use[i->first]==false)
            {
                use[i->first]=true;
                q.push(i->first);
            }
        }
    }
}
vector<int> v;
int d[20][20],st[20];
int main()
{
    in>>n>>m>>k;
    a=vector< vector< pair<int,int> > > (n+1);
    for(int i=2;i<=k+1;i++)
        in>>c[i];
    c[1]=1;
    c[k+2]=n;
    k+=2;
    for(;m;m--)
    {
        int x,y,z;
        in>>x>>y>>z;
        a[x].pb(mp(y,z));
        a[y].pb(mp(x,z));
    }
    for(int i=1;i<=k;i++)
    {
        st[i]=i;
        bellman(c[i],v);
        for(int j=1;j<=k;j++)
            d[i][j]=v[c[j]];
    }
    int sol=INF;
    do{
        int cost=0;
        for(int i=2;i<=k+2;i++)
            cost+=d[st[i-1]][st[i]];
        sol=min(cost,sol);
    }while(next_permutation(st+2,st+k));
    out<<sol<<'\n';
    return 0;
}