Cod sursa(job #2323091)

Utilizator PredaBossPreda Andrei PredaBoss Data 18 ianuarie 2019 20:04:19
Problema Ubuntzei Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.5 kb
#include <bits/stdc++.h>
#define pii pair<int,int>
#define piii pair<pii,int>
#define INF INT_MAX
using namespace std;
ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");
vector<pair<int,int> >nod[2005];
int n,m,k,x,y,c;
int pretin[20];
int val[2005];
int dp[2005][2005];
int ans[32800][20];
vector<int>elem[2005],fri;
priority_queue<pii,vector<pii >,greater<pii > >pq;
void dijkstra(int node)
{
    pq.push({0,node});
    while(!pq.empty())
    {
        int pos=pq.top().second;
        int v=pq.top().first;
        pq.pop();
        if(val[pos]!=INF) continue;
        val[pos]=v;
        for(int i=0;i<nod[pos].size();i++)
            if(val[nod[pos][i].first]==INF)
                pq.push({nod[pos][i].second+v,nod[pos][i].first});
    }
}
void get_ans()
{
    int lim=(1<<k)-1;
    for(int i=0;i<=lim;i++)
        for(int j=0;j<=k+1;j++)
            ans[i][j]=INF;
    for(int mask=1;mask<=lim;mask++)
        for(int i=1;i<fri.size()-1;i++)
            if(mask&val[fri[i]])
            {
                int mask2=mask-val[fri[i]];
                if(mask2==0)
                {
                    ans[mask][i]=dp[1][fri[i]];
                    continue;
                }
                for(int j=1;j<fri.size()-1;j++)
                    if(ans[mask2][j]!=INF)
                        ans[mask][i]=min(ans[mask][i],ans[mask2][j]+dp[fri[i]][fri[j]]);
            }
    int rez=INF;
    for(int i=1;i<fri.size()-1;i++)
        rez=min(rez,ans[lim][i]+dp[fri[i]][n]);
    fout<<rez;
}
int main()
{
    fin>>n>>m>>k;
    for(int i=0;i<k;i++)
        fin>>pretin[i];
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        nod[x].push_back({y,c});
        nod[y].push_back({x,c});
    }
    for(int j=0;j<=n;j++) val[j]=INF;
    dijkstra(1);
    for(int i=0;i<k;i++)
    {
        dp[1][pretin[i]]=val[pretin[i]];
        dp[pretin[i]][1]=val[pretin[i]];
    }
    dp[1][n]=val[n];
    dp[n][1]=val[n];
    if(k==0)
    {
        fout<<val[n];
        return 0;
    }
    for(int i=0;i<k;i++)
    {
        for(int j=0;j<=n;j++) val[j]=INF;
        dijkstra(pretin[i]);
        for(int j=0;j<k;j++)
            dp[pretin[i]][pretin[j]]=val[pretin[j]];
        dp[pretin[i]][n]=val[n];
        dp[n][pretin[i]]=val[n];
    }
    for(int j=0;j<=n;j++) val[j]=-1;
    fri.push_back(1);
    for(int i=0;i<k;i++)
    {
        val[pretin[i]]=1<<i;
        fri.push_back(pretin[i]);
    }
    fri.push_back(n);
    get_ans();
    return 0;
}