Cod sursa(job #2330971)

Utilizator TudorCaloianCaloian Tudor-Ioan TudorCaloian Data 29 ianuarie 2019 00:59:56
Problema Ubuntzei Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.47 kb
#include <bits/stdc++.h>
#define pi pair <int, int>
#define mp make_pair
#define f first
#define s second
#define inf (1<<30)
using namespace std;

ifstream fin("ubuntzei.in");
ofstream fout("ubuntzei.out");

int n, m, k, a[50], d[2005][2005], dp[(1<<16)][50];
bool city[2005], viz[10005];
vector < pair <int, pair <int, int> > > v[2005];
priority_queue <pi, vector<pi>, greater<pi> > pq;

void Dijkstra()
{
    for(int i = 0; i <= k; i++)
    {
            memset(viz, 0, sizeof(viz));
            pq.push(mp(0, a[i]));

          while(!pq.empty())
        {
            int nod = pq.top().s;
            int dist = pq.top().f;
            pq.pop();
            viz[nod] = 1;
            if(city[nod] && d[a[i]][nod] == 0)
            {
                d[a[i]][nod] = dist;
                d[nod][a[i]] = dist;


            }


            for(int j = 0; j < v[nod].size(); j++)
            {
                int child = v[nod][j].f;
                int price = v[nod][j].s.f;
                int edge = v[nod][j].s.s;

                if(!viz[child])
                {
                    pq.push(mp(dist+price, child));

                }

            }
        }
      }


}

int main()
{
    fin >> n >> m;
    fin >> k;
    for(int i = 1; i <= k; i++)
    {
        int x;
        fin >> x;
        city[x] = 1;
        a[i] = x;
    }

    for(int i = 1; i <= m; i++)
    {
        int a, b, c;
        fin >> a >> b >> c;
        v[a].push_back(make_pair(b, mp(c, i)));
        v[b].push_back(make_pair(a, mp(c, i)));
    }

    a[0] = 1;
    city[1] = 1;
    city[n] = 1;
    a[k+1] = n;
    Dijkstra();
    if(k == 0)
    {
        fout << d[1][n];
        return 0;
    }
    for(int i = 0; i < (1<<k); i++)
        for(int j = 0; j <= 16; j++)
            dp[i][j] = inf;
    for(int i = 0; i < k; i++)
        dp[(1<<i)][i+1] = d[1][a[i+1]];

    for(int i = 0; i < (1<<k)-1; i++)
    {
        for(int j = 0; j < k; j++)
            if(((1<<j)&i) != 0)
                for(int p = 0; p < k; p++)
                    if(((1<<p)&i) == 0)
                     {
                         dp[(i|(1<<p))][p+1] = min(dp[(i|(1<<p))][p+1], dp[i][j+1]+d[a[j+1]][a[p+1]]);

                     }

    }
    int ans = inf;

    for(int i = 1; i <= k; i++)
        ans = min(ans, dp[(1<<k)-1][i]+d[a[i]][n]);
    fout << ans << '\n';
    //fout << d[1][2] << '\n';
    //fout << d[2][4];


    return 0;
}