Cod sursa(job #2721894)

Utilizator Tudor_PascaTudor Pasca Tudor_Pasca Data 12 martie 2021 13:39:25
Problema Ubuntzei Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.19 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int INF = 1e9;

int n, m, k, ans = INF;
int adjMat[2010][2010], loc[20], hamilton[262144][18];
vector<vector<pair<int, int>>> adj;

void royFloyd()
{
    for(int x = 1; x <= n; x++)
        for(int i = 1; i <= n; i++)
            for(int j = 1; j <= n; j++)
                adjMat[i][j] = min(adjMat[i][j], adjMat[i][x] + adjMat[x][j]);
}

void buildHamilton()
{
    adj.resize(k + 3);

    for(int i = 1; i <= k; i++)
        for(int j = 1; j <= k; j++)
            if(i != j)
                adj[i].push_back({j, adjMat[loc[i]][loc[j]]});

    for(int i = 1; i <= k; i++)
        hamilton[(1 << i)][i] = 0;

    /*
    for(int i = 1; i <= k; i++)
        cout << i << ' ';

    cout << '\n';

    for(int i = 1; i <= k; i++)
        cout << loc[i] << ' ';

    cout << '\n';

    for(int i = 1; i <= k; i++)
    {
        cout << "NODE: " << i << '\n';
        for(auto it: adj[i])
            cout << it.first << ' ' << it.second << '\n';
        cout << "\n";
    }
    */

    for(int i = 1; i < (1 << k); i++)
    {
        for(int j = 1; j <= k; j++)
        {
            if(i & (1 << j) == 0)
                continue;

            for(auto it: adj[j])
                if(i & (1 << it.first))
                    hamilton[i][j] = min(hamilton[i][j], hamilton[i ^ (1 << it.first)][it.first] + it.second);
        }
    }
}

int main()
{
    in >> n >> m >> k;

    for(int i = 1; i <= k; i++)
        in >> loc[i];

    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
            if(i != j)
                adjMat[i][j] = INF;

    for(int i = 1; i <= m; i++)
    {
        int x, y, cost;
        in >> x >> y >> cost;

        adjMat[x][y] = cost;
        adjMat[y][x] = cost;
    }

    royFloyd();

    buildHamilton();

    for(int x = 1; x <= k; x++)
        for(int i = 1; i <= k; i++)
            for(int j = 1; j <= k; j++)
                ans = min(ans, adjMat[1][loc[i]] + adjMat[loc[j]][n] + hamilton[x][(1 << k) - 1]);

    out << ans << '\n';

    return 0;
}