骑士走键盘

2020-02-20  本文已影响0人  别时茫茫

骑士走键盘

标签(空格分隔): algorithm


骑士在键盘上走日形状的个数

next_=\left\{\begin{matrix} [0:[4, 6]]\\ [1:[6, 8 ]]\\ [2:[7, 9]]\\ [3:[4, 8 ]]\\ [4:[0, 3, 9 ]]\\ [5:[ ]]\\ [6:[0, 1, 7 ]]\\ [7:[2, 6]]\\ [8:[1, 3]]\\ [9:[2, 4]]\\ \end{matrix}\right.


方法一:递归

f(k,i,n) = \sum_{j=0,a=next[k]}^{len(a)}{f(a_j,i+1,n)}

    unordered_map<int, vector<int>> umap;
    const int M = int(1e9 + 7);
    int knightDialer(int N)
    {
        umap[0] = { 4, 6 };
        umap[1] = { 6, 8 };
        umap[2] = { 7, 9 };
        umap[3] = { 4, 8 };
        umap[4] = { 0, 3, 9 };
        umap[5] = {};
        umap[6] = { 0, 1, 7 };
        umap[7] = { 2, 6 };
        umap[8] = { 1, 3 };
        umap[9] = { 2, 4 };
        int c = 0;
        for (int i = 0; i < 10; i++) {
            c += knight(i, 1, N);
            c %= M;
        }
        return c;
    }

    int knight(int k, int step, int n)
    {
        if (step == n)
            return 1;
        auto list = umap[k];
        int size = list.size();
        int s = 0;
        for (int i = 0; i < size; i++) {
            int x = knight(list[i], step + 1, n);
            s += x % M;
            s %= M;
        }
        return s % M;
    }

方法二:转化为备忘录递归(自顶向下的方法)

unordered_map<int, vector<int>> umap;
    unordered_map<string, int> cache;
    const int M = int(1e9 + 7);
    int knightDialer(int N)
    {
        umap.clear();
        cache.clear();
        umap[0] = { 4, 6 };
        umap[1] = { 6, 8 };
        umap[2] = { 7, 9 };
        umap[3] = { 4, 8 };
        umap[4] = { 0, 3, 9 };
        umap[5] = {};
        umap[6] = { 0, 1, 7 };
        umap[7] = { 2, 6 };
        umap[8] = { 1, 3 };
        umap[9] = { 2, 4 };
        int c = 0;
        for (int i = 0; i < 10; i++) {
            c += knight(i, 1, N);
            c %= M;
        }
        return c;
    }

    int knight(int k, int step, int n)
    {
        if (step == n)
            return 1;
        string key = to_string(k) + "-" + to_string(step);
        if (cache.find(key) != cache.end()) {
            return cache[key];
        }
        auto list = umap[k];
        int size = list.size();
        int s = 0;
        for (int i = 0; i < size; i++) {
            int x = knight(list[i], step + 1, n);
            s += x % M;
            s %= M;
        }
        cache[key] = s % M;
        return cache[key];
    }


方法三:转化为备忘录递归(自底向上的方法)(转化为迭代计算)

f[i,n] = \sum_{k=0,a=next(i)}^{len(a)}f[a[k],n-1]

unordered_map<int, vector<int>> umap;
    const int M = int(1e9 + 7);
    int knightDialer(int N){
        umap[0] = { 4, 6 };
        umap[1] = { 6, 8 };
        umap[2] = { 7, 9 };
        umap[3] = { 4, 8 };
        umap[4] = { 0, 3, 9 };
        umap[5] = {};
        umap[6] = { 0, 1, 7 };
        umap[7] = { 2, 6 };
        umap[8] = { 1, 3 };
        umap[9] = { 2, 4 };

        int c = 0;
        int size = 10;

        vector<vector<int>> dp(size, vector<int>(N, 0));
        // base case #1
        for (int i = 0; i < size; i++) {
            dp[i][0] = 1;
        }

        for (int s = 1; s < N; s++) {
            for (int i = 0; i < size; i++) {
                auto next = umap[i];
                for (auto e : next) { //
                    dp[i][s] += dp[e][s - 1];
                    dp[i][s] %= M;
                }
            }
        }
        for (int i = 0; i < size; i++) {
            c += dp[i][N - 1];
            c %= M;
        }
        return c % M;
    }
上一篇下一篇

猜你喜欢

热点阅读