474. 一和零
在计算机界中,我们总是追求用有限的资源获取最大的收益。
现在,假设你分别支配着 m 个 0 和 n 个 1。另外,还有一个仅包含 0 和 1 字符串的数组。
你的任务是使用给定的 m 个 0 和 n 个 1 ,找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。
注意:
给定 0 和 1 的数量都不会超过 100。
给定字符串数组的长度不会超过 600。
示例 1:
输入: Array = {"10", "0001", "111001", "1", "0"}, m = 5, n = 3
输出: 4
解释: 总共 4 个字符串可以通过 5 个 0 和 3 个 1 拼出,即 "10","0001","1","0" 。
示例 2:
输入: Array = {"10", "0", "1"}, m = 1, n = 1
输出: 2
解释: 你可以拼出 "10",但之后就没有剩余数字了。更好的选择是拼出 "0" 和 "1" 。
题解
本题是01背包问题的多维度扩展版本
01背包限制只有背包的容量。所以我们的dp数组第二维度是容量j。
这个问题对于每个元素同样与取和不取两种。但是限制不在是一个,而是两个,1不能超过m,0不能超过n。
那么我们的dp数组为 d p [ i ] [ j ] [ k ] dp[i][j][k] dp[i][j][k],记录前i个元素,在1的数量为j,0的数量为k下的最大收益。
我们的初始条件为
d
p
[
∗
]
[
0
]
[
0
]
=
0
dp[*][0][0]=0
dp[∗][0][0]=0即一个元素都不选,且1的数量、0的数量都为0时,我们的最大收益为0.
d
p
[
i
]
[
j
]
[
k
]
=
max
{
d
p
[
i
−
1
]
[
j
−
n
u
m
s
1
]
[
k
−
n
u
m
s
0
]
+
1
选
择
该
字
符
串
d
p
[
i
−
1
]
[
j
]
[
k
]
不
选
择
该
字
符
串
dp[i][j][k] = \max \begin{cases} dp[i-1] [j-nums1] [k-nums0] +1 \quad 选择该字符串 \\ dp[i-1] [j] [k] \quad 不选择该字符串 \end{cases}
dp[i][j][k]=max{dp[i−1][j−nums1][k−nums0]+1选择该字符串dp[i−1][j][k]不选择该字符串
在经过数组维度优化,可以将三维数组优化到二维数组:
最终方案:
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
int len = strs.length;
int [][] dp = new int[m+1][n+1];
//初始条件
dp[0][0] = 0;
for(int i = 0; i<strs.length; i++){
int nums1 = count1(strs[i]);
int nums0 = strs[i].length()-nums1;
for(int j = m; j-nums0>=0; j--){
for(int k = n; k-nums1>=0; k--){
dp[j][k] = Math.max(dp[j-nums0][k-nums1]+1, dp[j][k]);
}
}
}
return dp[m][n];
}
private int count1(String str){
int ans = 0;
for(int i=0; i<str.length(); i++){
if(str.charAt(i)=='1'){
ans++;
}
}
return ans;
}
}