HDU 3001 Travelling 状态DP

TSP问题，不懂就是每个点最多访问两次，最少访问一次。

所以，我们可以用三进制来当做状态。

这个题练习了一下三进制……0、1、2

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <functional>
#include <time.h>

using namespace std;

const int INF = 1<<30;

int Matrix[15][15];
int base[70000][15]; //base[i][j]表示 数字i从右边第j位起的三进制值
int dp[70000][15];
int thr[20];
int n, m;

inline bool check(int x) { //判断是否是最终状态
    for (int i = 0; i < n; i++) if (!base[x][i]) //如果某一位是0则不是
        return false;
    return true;
}

inline void update(int st, int u, int v) { //更新状态
    int &now = dp[st+thr[v]][v];
    if (now<0) now = dp[st][u]+Matrix[u][v];
    else now = min(now, dp[st][u]+Matrix[u][v]);
}

void init() {
    thr[0] = 1;
    for (int i = 1; i < 20; i++)
        thr[i] = thr[i-1]*3;
    for (int i = 0; i < 70000; i++) {
        int j = i, k = 0;
        while (j) {
            base[i][k++] = j%3;
            j /= 3;
        }
    }
}

void input() {
    memset(Matrix, -1, sizeof(Matrix));
    for (int i = 0; i < m; i++) {
        int x, y, c;
        scanf("%d%d%d", &x, &y, &c);
        x--; y--;
        if (-1==Matrix[x][y]) Matrix[x][y] = Matrix[y][x] = c;
        else Matrix[x][y] = Matrix[y][x] = min(Matrix[x][y], c);
    }
}

void solve() {
    memset(dp, -1, sizeof(dp));
    for (int i = 0; i < n; i++)
        dp[thr[i]][i] = 0;
    for (int i = 1; i < thr[n]; i++) {
        for (int u = 0; u < n; u++) if (dp[i][u]>-1) { //此状态可达
            for (int v = 0; v < n; v++) if (u!=v && base[i][v]<2 && Matrix[u][v]>-1){ //可以转移到下一个状态
                update(i, u, v); //更新下一个状态
            }
        }
    }

    int ans = -1;
    for (int i = 0; i < thr[n]; i++) if (check(i)) {
        for (int j = 0; j < n; j++) if (-1 < dp[i][j]) {
            ans = ans>-1 ? min(ans, dp[i][j]) : dp[i][j];
        }
    }
    printf("%d
", ans);
}

int main() {
    #ifdef Phantom01
        freopen("HDU3001.txt", "r", stdin);
    #endif //Phanaom01

    init();
    while (scanf("%d%d", &n, &m)!=EOF) {
        input();
        solve();
    }

    return 0;
}