Cod sursa(job #2546322)

Utilizator memecoinMeme Coin memecoin Data 14 februarie 2020 08:22:53
Problema Ciclu hamiltonian de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.06 kb
#include <fstream>
#include <iomanip>
#include <string>
#include <stdio.h>
#include <vector>
#include <algorithm>
#include <math.h>
#include <set>
#include <map>
#include <string.h>
#include <queue>
#include <stack>

#define INF 0x3f3f3f3f

using namespace std;

#ifdef DEBUG
string name = "data";
#else
string name = "hamilton";
#endif

ifstream fin(name + ".in");
ofstream fout(name + ".out");

#define MAXN 19

int n, m;

int c[MAXN][MAXN];
int best[1 << MAXN][MAXN];

void gen(int k, int bit, int nrOne, vector<int> &result) {
    if (bit > (n - nrOne)) {
        return;
    }
    
    if (nrOne == 0) {
        result.push_back(k);
        return;
    }
    
    gen(k | (1 << bit), bit + 1, nrOne - 1, result);
    gen(k, bit + 1, nrOne, result);
}

vector<int> genNr(int nrOne) {
    vector<int> result;
    
    if (nrOne == 0) {
        result.push_back(0);
        return result;
    }
    
    gen(0, 0, nrOne, result);
    
    return result;
}

int main() {
    
    fin >> n >> m;
    
    memset(c, 0x3f, sizeof(c));
    
    for (int i = 0; i < m; ++i) {
        int x,y,z;
        fin >> x >> y >> z;
        c[x][y] = z;
    }
    
    memset(best, 0x3f, sizeof(best));
    int x = 1;
    
    for (int i = 0; i < n; ++i) {
        best[x][i] = 0;
        x *= 2;
    }
    
    for (int i = 2; i <= n; ++i) {
        auto v = genNr(i);
        
        for (int x: v) {
            for (int bit = 0; bit < n; ++bit) {
                for (int j = 0; j < n; ++j) {
                    if (!(x & (1 << j))) {
                        continue;
                    }
                    
                    if (c[j][bit] != INF) {
                        best[x][bit] = min(best[x][bit], best[x & (~(1 << bit))][j] + c[j][bit]);
                    }
                }
            }
        }
    }
    
    int bestest = INF;
    
    for (int i = 1; i < n; ++i) {
        if (c[0][i] != INF) {
            bestest = min(bestest, best[(1 << n) - 1][i] + c[i][0]);
        }
    }
    
    if (bestest == INF) {
        fout << -1;
    } else {
        fout << bestest;
    }
    
    return 0;
}