思路分析:
-
初始化数组
arr
: 遍历输入的字符串数组strs
,对每个字符串统计0和1的数量,分别保存在arr[i][0]
和arr[i][1]
中。这样得到了一个二维数组arr
,其中arr[i][0]
表示第i个字符串中0的数量,arr[i][1]
表示第i个字符串中1的数量。 -
动态规划数组
dp
初始化: 定义一个二维动态规划数组dp
,其中dp[i][j]
表示使用i个1和j个0时的最大字符串数量。初始化dp
为全零。 -
遍历字符串数组更新
dp
数组: 遍历字符串数组,对于每个字符串,使用动态规划的思想更新dp
数组。具体地,从n
到arr[k][1]
(字符串中1的数量),从m
到arr[k][0]
(字符串中0的数量)逆序遍历dp
数组,更新dp[i][j]
为当前值与选择或不选择当前字符串的最大值。 -
返回结果: 最终返回
dp[n][m]
,即使用给定的m
个0和n
个1时的最大字符串数量。
class Solution {
public:
int findMaxForm(vector<string>& strs, int m, int n) {
// 定义数组arr[i][0]表示第i个字符串中0的数量,arr[i][1]表示第i个字符串中1的数量
int arr[strs.size()][2];
// 遍历字符串数组,初始化arr数组
for(int i = 0; i < strs.size(); i++) {
arr[i][0] = 0;
arr[i][1] = 0;
// 统计当前字符串中0和1的数量
for(int j = 0; j < strs[i].size(); j++) {
if(strs[i][j] == '0')
arr[i][0]++;
else
arr[i][1]++;
}
}
// 定义二维动态规划数组dp,dp[i][j]表示使用i个0和j个1时的最大字符串数量
vector<vector<int>> dp(n + 1, vector<int>(m + 1, 0));
// 遍历字符串数组,更新动态规划数组dp
for(int k = 0; k < strs.size(); k++) {
for(int i = n; i >= arr[k][1]; i--) {
for(int j = m; j >= arr[k][0]; j--) {
// 更新dp数组,选择当前字符串或者不选择当前字符串,取两者中的最大值
dp[i][j] = max(dp[i][j], dp[i - arr[k][1]][j - arr[k][0]] + 1);
}
}
}
// 返回dp数组中最大的字符串数量,即dp[n][m]
return dp[n][m];
}
};