Cod sursa(job #1617805)

Utilizator bogdanboboc97Bogdan Boboc bogdanboboc97 Data 27 februarie 2016 16:27:38
Problema Ubuntzei Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.05 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],dp[19][19][139001];
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]];
    }
    for(int i=2;i<=k;i++)
        dp[1][i][1<<i]=d[1][i];
    for(int i=1;i<k;i++)
    {
        for(int p=0;p<=(1<<(k+1));p++)
        for(int j=2;j<=k;j++)
        if(dp[i][j][p])
        {
            for(int t=2;t<=k;t++)
                if((p&(1<<t))==0)
                dp[i+1][t][p or (1<<t)]=((dp[i+1][t][p or (1<<t)] > dp[i][j][p]+d[j][t] || dp[i+1][t][p or (1<<t)]==0) ? dp[i][j][p]+d[j][t] : dp[i+1][t][p or (1<<t)]);
        }
    }
    int sol=INF;
    for(int i=0;i<=(1<<k);i++)
        if(dp[k][k][i])
        sol=min(sol,dp[k][k][i]);
    out<<sol<<'\n';
    return 0;
}