Cod sursa(job #3174783)

Utilizator amcbnCiobanu Andrei Mihai amcbn Data 25 noiembrie 2023 10:05:09
Problema Ciclu hamiltonian de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.11 kb
#include <bits/stdc++.h>
#include <random>
#include <chrono>
using namespace std;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const char nl = '\n';
const char sp = ' ';
const int inf = 0x3f3f3f3f;
const long long INF = 1000000000000000000;
const int mod = 1e9 + 7;
const char out[2][4]{ "NO", "YES" };
#define all(a) a.begin(), a.end()
using ll = long long;
ifstream fin("hamilton.in");
ofstream fout("hamilton.out");

#define variableName(var) #var
#define __debug(var) cout << #var << " = " << var << '\n'
inline int rint(int a, int b) { return uniform_int_distribution<int>(a, b)(rng); }

const int nmax = 18;
int n, m;
vector<pair<int, int>> g[nmax];
int dp[nmax][1 << nmax];

/*

dp[j][mask] = incep in 0, sunt in j, am vizitat tot ce e in mask
dp[k][mask ^ (1 << k)] = dp[i][j][mask] + c
daca exista muchie de greutate c intre j si k
si daca k nu e inclus deja in mask

*/

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    fin >> n >> m;
    for (int i = 0; i < m; ++i) {
        int u, v, w;
        fin >> u >> v >> w;
        g[u].push_back({ v, w });
    }
    for (int j = 0; j < n; ++j) {
        for (int mask = 0; mask < (1 << n); ++mask) {
            dp[j][mask] = inf;
        }
    }
    dp[0][1] = 0;
    for (int mask = 0; mask < (1 << n); ++mask) {
        for (int j = 0; j < n; ++j) {
            if ((mask >> j) & 1) {
                if (dp[j][mask] != inf) {
                    for (auto& e : g[j]) {
                        int k = e.first, w = e.second;
                        if (!((mask >> k) & 1)) {
                            dp[k][mask ^ (1 << k)] = min(dp[k][mask ^ (1 << k)], dp[j][mask] + w);
                        }
                    }
                }
            }
        }
    }
    int ans = inf;
    for (int i = 0; i < n; ++i) {
        for (auto& e : g[i]) {
            if (e.first == 0) {
                ans = min(ans, dp[i][(1 << n) - 1] + e.second);
            }
        }
    }
    if (ans == inf) {
        fout << "Nu exista solutie";
    }
    else {
        fout << ans;
    }
}