Cod sursa(job #2300953)

Utilizator AlexandruabcdeDobleaga Alexandru Alexandruabcde Data 12 decembrie 2018 14:05:05
Problema Ciclu hamiltonian de cost minim Scor 75
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.43 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f ("hamilton.in");
ofstream g ("hamilton.out");

constexpr int INF = 0x3F3F3F3F;
constexpr int NMAX=18, MMAX=262145;

vector <int> G[NMAX];

int n, m, x, y, c;

int cost[NMAX][NMAX];

int dp[NMAX][MMAX];

inline void citire ()
{
    f >> n >> m;

    for (int i = 0; i < n; ++i)
    {
        for (int j = 0; j < n; ++j)
            cost[i][j] = INF;
    }

    for (int i = 1; i <= m; ++i)
    {
        f >> x >> y >> c;

        G[y].push_back(x);

        cost[x][y] = c;
    }

    for (int i = 0; i < NMAX; ++i)
    {
        for (int j = 0; j < MMAX; ++j)
        {
            dp[i][j] = -1;
        }
    }

    dp[0][1] = 0;
}

inline int functie (int x, int S)
{
    if (dp[x][S] == -1) {

        ///S - X
        ///x - Y
        dp[x][S] = INF;

        for (int k = 0; k < G[x].size(); ++k)
            if (S & (1<<G[x][k]))
                dp[x][S] = min(dp[x][S], cost[G[x][k]][x] + functie(G[x][k], S ^ (1<<x)));
    }

    return dp[x][S];
}

inline void solve()
{
    int sol = INF;

    for (int i = 0; i < G[0].size(); ++i)
        sol = min(sol, functie(G[0][i], (1<<n) - 1) + cost[G[0][i]][0]);

    if (sol == INF)
    {
        g <<"Nu exista solutie" << '\n';

        return;
    }

    g << sol << '\n';
}

int main()
{
    f.tie(NULL);

    citire();

    solve();

    return 0;
}