一.题目要求
给你一个下标从 0 开始的 m x n 整数矩阵 grid 和一个整数 k 。你从起点 (0, 0) 出发,每一步只能往 下 或者往 右 ,你想要到达终点 (m - 1, n - 1) 。
请你返回路径和能被 k 整除的路径数目,由于答案可能很大,返回答案对 109 + 7 取余 的结果。
二.题目难度
1952
三.输入样例
示例 1:
输入:grid = [[5,2,4],[3,0,5],[0,7,2]], k = 3
输出:2
解释:有两条路径满足路径上元素的和能被 k 整除。
第一条路径为上图中用红色标注的路径,和为 5 + 2 + 4 + 5 + 2 = 18 ,能被 3 整除。
第二条路径为上图中用蓝色标注的路径,和为 5 + 3 + 0 + 5 + 2 = 15 ,能被 3 整除。
示例 2:
输入:grid = [[0,0]], k = 5
输出:1
解释:红色标注的路径和为 0 + 0 = 0 ,能被 5 整除。
示例 3:
输入:grid = [[7,3,4,9],[2,3,6,2],[2,3,7,0]], k = 1
输出:10
解释:每个数字都能被 1 整除,所以每一条路径的和都能被 k 整除。
四.解题思路
五.代码实现
class Solution {
public:
int numberOfPaths(vector<vector<int>>& grid, int k) {
int m = grid.size();
int n = grid[0].size();
#define mod 1000000007
vector<vector<vector<int>>> dp(m, vector<vector<int>>(n, vector<int>(k, 0)));
dp[0][0][grid[0][0] % k] = 1;
for(int i = 0; i < m; i++)
{
for(int j = 0; j < n; j++)
{
if(!i && !j)
continue;
if(!i)
{
for(int a = 0; a < k; a++)
dp[i][j][(a + grid[i][j]) % k] = dp[i][j - 1][a] % mod;
}
else if(!j)
{
for(int a = 0; a < k; a++)
dp[i][j][(a + grid[i][j]) % k] = dp[i - 1][j][a] % mod;
}
else
{
for(int a = 0; a < k; a++)
dp[i][j][(a + grid[i][j]) % k] = (dp[i - 1][j][a] + dp[i][j - 1][a]) % mod;
}
}
}
return dp[m - 1][n - 1][0];
}
};