Cod sursa(job #2323068)

Utilizator PredaBossPreda Andrei PredaBoss Data 18 ianuarie 2019 19:33:18
Problema Ubuntzei Scor 65
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.42 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];
vector<int>elem[2005],fri;
bitset<32800>ap[2005];
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()
{
    priority_queue<piii,vector<piii >,greater<piii > >q;
    q.push({{0,0},1});
    while(!q.empty())
    {
        int pos=q.top().second;
        int cst=q.top().first.first;
        int v=q.top().first.second;
        q.pop();
        if(ap[pos][v]) continue;
        ap[pos][v]=1;
        if(pos==n && v==(1<<k)-1)
        {
            fout<<cst;
            return;
        }
        for(int i=0;i<fri.size();i++)
        {
            if(pos==fri[i]) continue;
            int counter=v;
            if(val[fri[i]]!=-1 && (v&(val[fri[i]]))==0)
                counter+=val[fri[i]];
            if(ap[fri[i]][counter]) continue;
            q.push({{cst+dp[fri[i]][pos],counter},fri[i]});
        }
    }
}
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];
    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;
}