题目:
在计算机界中,我们总是追求用有限的资源获取最大的收益。
现在,假设你分别支配着 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" 。
二维费用背包问题,第一遍 时间复杂度太高了
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
if(strs == null || strs.length == 0)
return 0;
int[][] dp = new int[m+1][n+1];
for(int i = 0; i < strs.length; i++){
for(int j = m;j >= zero(strs[i]); j--){
for(int k = n; k >= one(strs[i]); k--){
dp[j][k] = Math.max(dp[j][k],dp[j-zero(strs[i])][k-one(strs[i])]+1);
}
}
}
return dp[m][n];
}
public int zero(String s){
int count = 0;
for(int i = 0; i < s.length(); i++){
if(s.charAt(i) == '0')
count++;
}
return count;
}
public int one(String s){
return s.length()-zero(s);
}
}
第二遍把计算每个字符串zero和one的数量放在第一重循环里写,效率显著提高
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
if(strs == null || strs.length == 0)
return 0;
int[][] dp = new int[m+1][n+1];
for(int i = 0; i < strs.length; i++){
int zero = 0;int one = 0;
for(char c : strs[i].toCharArray()){
if(c == '0')
zero++;
else
one++;
}
for(int j = m;j >= zero; j--){
for(int k = n; k >= one; k--){
dp[j][k] = Math.max(dp[j][k],dp[j-zero][k-one]+1);
}
}
}
return dp[m][n];
}
}