【三维 DP 的简单应用】

慕容青峰

已于 2022-12-05 10:05:30 修改

阅读量503

点赞数

于 2022-12-01 23:35:02 首次发布

本文链接：https://blog.csdn.net/m0_57883655/article/details/128140691

版权

动态规划专栏收录该内容

9 篇文章 0 订阅

订阅专栏

LeetCode 2435. 矩阵中和能被 K 整除的路径

算法：动态规划

设 $f (i, j, t)$ 表示当前在的格子坐标为 $(i, j)$ ，且所有到达此点的路径和（包括此点）对 $k$ 取余数为 $t$ 的方案数。

由于只能往下和往右走，因此可以得到状态转移方程：

$\% k) = f(i - 1, j, t) + f(i, j - 1, t)$

所以只需要三重循环，前两重枚举坐标，最内层枚举余数，从 0 ~ k - 1 即可

时间复杂度 $O (n m k)$

C++ 代码

typedef long long LL;

const int MOD = 1e9 + 7;

class Solution {
public:
    int numberOfPaths(vector<vector<int>>& g, int k) {
        int n = g.size(), m = g[0].size();

        vector<vector<vector<int>>> f(n, vector<vector<int>>(m, vector<int>(k)));
        f[0][0][g[0][0] % k] = 1;

        for (int i = 0; i < n; i ++ )
            for (int j = 0; j < m; j ++ ) {
                if (i) {
                    for (int t = 0; t < k; t ++ ) {
                        auto &x = f[i][j][(g[i][j] + t) % k];
                        x = ((LL)x + f[i - 1][j][t]) % MOD;
                    }
                }
                if (j) {
                    for (int t = 0; t < k; t ++ ) {
                        auto &x = f[i][j][(g[i][j] + t) % k];
                        x = ((LL)x + f[i][j - 1][t]) % MOD;
                    }
                }
            }

        return f[n - 1][m - 1][0];
    }
};