Cod sursa(job #2322694)

Utilizator PredaBossPreda Andrei PredaBoss Data 18 ianuarie 2019 09:50:32
Problema A+B Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.14 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 > >pq;
    pq.push({{0,0},1});
    while(!pq.empty())
    {
        int pos=pq.top().second;
        int cst=pq.top().first.first;
        int v=pq.top().first.second;
        pq.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;
        }
    }
}
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});
    }
    memset(val,INF,sizeof val);
    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++)
    {
        memset(val,INF,sizeof val);
        dijkstra(pretin[i]);
        for(int j=0;j<k;j++)
            dp[pretin[i]][pretin[j]]=val[pretin[j]];
        dp[pretin[i]][n]=val[n];
    }
    fri.push_back(1);
    for(int i=0;i<k;i++)
    {
        val[pretin[i]]=i;
        fri.push_back(pretin[i]);
    }
    fri.push_back(n);
    get_ans();
    return 0;
}