目录
1444. 切披萨的方案数
题目描述:
实现代码与解析:
二维后缀和 + 动态规划
原理思路:
1444. 切披萨的方案数
题目描述:
给你一个 rows x cols
大小的矩形披萨和一个整数 k
,矩形包含两种字符: 'A'
(表示苹果)和 '.'
(表示空白格子)。你需要切披萨 k-1
次,得到 k
块披萨并送给别人。
切披萨的每一刀,先要选择是向垂直还是水平方向切,再在矩形的边界上选一个切的位置,将披萨一分为二。如果垂直地切披萨,那么需要把左边的部分送给一个人,如果水平地切,那么需要把上面的部分送给一个人。在切完最后一刀后,需要把剩下来的一块送给最后一个人。
请你返回确保每一块披萨包含 至少 一个苹果的切披萨方案数。由于答案可能是个很大的数字,请你返回它对 10^9 + 7 取余的结果。
示例 1:
输入:pizza = ["A..","AAA","..."], k = 3 输出:3 解释:上图展示了三种切披萨的方案。注意每一块披萨都至少包含一个苹果。
示例 2:
输入:pizza = ["A..","AA.","..."], k = 3 输出:1
示例 3:
输入:pizza = ["A..","A..","..."], k = 1 输出:1
提示:
1 <= rows, cols <= 50
rows == pizza.length
cols == pizza[i].length
1 <= k <= 10
-
pizza
只包含字符'A'
和'.'
。
实现代码与解析:
二维后缀和 + 动态规划
class Solution {
public:
int ways(vector<string>& pizza, int k) {
int m = pizza.size(), n = pizza[0].size(), mod = 1e9 + 7;
vector<vector<vector<int>>> f(k, vector<vector<int>>(m, vector<int>(n)));
vector<vector<int>> apples(m + 1, vector<int>(n + 1)); // 后缀和
// 后缀和 与 初始化dp数组
for (int i = m - 1; i >= 0; i--)
{
for (int j = n - 1; j >= 0; j--)
{
apples[i][j] = apples[i + 1][j] + apples[i][j + 1] - apples[i + 1][j + 1] + (pizza[i][j] == 'A');
f[0][i][j] = apples[i][j] > 0;
}
}
for (int kk = 1; kk < k; kk++)
{
for (int i = 0; i < m; i++)
{
for (int j = 0; j < n; j++)
{
// 选择此刀的切割位置
// 水平切, 遍历切的位置
for (int a = i + 1; a < m; a++)
{
// 上面的一块中至少要有一个苹果
if (apples[i][j] > apples[a][j])
{
f[kk][i][j] = (f[kk][i][j] + f[kk - 1][a][j]) % mod;
}
}
// 垂直切
for (int b = j + 1; b < n; b++)
{
// 左侧块中至少有一个苹果
if (apples[i][j] > apples[i][b])
{
f[kk][i][j] = (f[kk][i][j] + f[kk - 1][i][b]) % mod;
}
}
}
}
}
return f[k - 1][0][0];
}
};
原理思路:
apples 数组,后缀和用于记录一块披萨中的苹果数量,用一块中的左上角来代替此块含有的苹果数。
此题的关键是,dp[ k ][ i ][ j ] 的含义:代表还剩下 k 刀没切,剩下的是左上角为 i ,j 的披萨状态时的切割方案总数。这是我自己的理解,力扣上dp数组定义的含义感觉不如我这样写和解释更直观,不过原理肯定是一样的。
知道dp数组的含义,就很好写了。
首先计算 apples 数组,这个就不用解释了,不会的话,建议去学习一下前缀和,二维前缀和的基础算法就行,同时初始化一下dp。
初始化dp数组:显然在还需要切0刀,剩下最后一块披萨中有苹果时,表示切好了,是一种情况,赋值为1,否则不成立赋值为0;
f[0][i][j] = apples[i][j] > 0;
遍历顺序:一定是先遍历切割刀数,因为就比如一个形状披萨状态下,切两刀肯定需要切一刀的状态递推而来,后面根据递推式也能看出来。
递推方程:两种切法分类讨论:
水平切:肯定是从第一行下边开始切,总不能切空气吧,所以是 i + 1 开始遍历,然后切完后上面的那块中一定要有苹果,所以需要判断一下,切完此刀后,剩下的大块需要再切 kk - 1刀,我们就不用再去遍历了,dp数组含义就是这个,根据这个写出递推式。
递推式:f[ kk ][ i ][ j ] = (f[ kk ][ i ][ j ] + f[ kk - 1 ][ a ][ j ]) % mod;
垂直切:与水平切同理,直接给出递推式:
递推式:f[ kk ][ i ][ j ] = (f[ kk ][ i ][ j ] + f[ kk - 1 ][ i ][ b ]) % mod;
最后,返回结果,显然,在初始状态还剩切k - 1刀时是我们需要的结果状态。
return f[ k - 1 ][ 0 ][ 0 ]; 文章来源:https://www.toymoban.com/news/detail-657081.html
结束。文章来源地址https://www.toymoban.com/news/detail-657081.html
到了这里,关于Leetcode每日一题:1444. 切披萨的方案数(2023.8.17 C++)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!