在计算机界中,我们总是追求用有限的资源获取最大的收益。
现在,假设你分别支配着 m
个0
和 n
个 1
。另外,还有一个仅包含 0
和 1
字符串的数组。
你的任务是使用给定的 m
个 0
和 n
个 1
,找到能拼出存在于数组中的字符串的最大数量。每个0
和 1
至多被使用一次。
注意:
给定
0
和1
的数量都不会超过100
。
给定字符串数组的长度不会超过600
。
思路:
01
背包的变换问题,不同的是在背包问题中,我们只有一种容量,而在这道题中,我们有0
和1
两种容量。每个物品(字符串)需要分别占用0
和1
的若干容量,并且所有物品的价值均为1
。
class Solution {
public int findMaxForm(String[] strs, int m, int n) {
int size = strs.length;
int z = m + 1;//zero
int o = n + 1;//one
int[][][] dp = new int[size][z][o];
int[] cnt = getZerosAndOnes(strs[0]);
for (int i = 0; i < z; i++) {
for (int j = 0; j < o; j++) {
if (i >= cnt[0] && j >= cnt[1])
dp[0][i][j] = 1;
}
}
for (int s = 1; s < size; s++) {
for (int i = 0; i < z; i++) {
for (int j = 0; j < o; j++) {
cnt = getZerosAndOnes(strs[s]);
if (i - cnt[0] >= 0 && j - cnt[1] >= 0)
dp[s][i][j] = Math.max(dp[s - 1][i][j], 1+dp[s - 1][i - cnt[0]][j - cnt[1]]);
else
dp[s][i][j] = dp[s-1][i][j];
}
}
}
return dp[size - 1][m][n];
}
private int[] getZerosAndOnes(String str) {
int[] res = new int[2];
for (int i = 0; i < str.length(); i++) {
if (str.charAt(i) == '1')
res[1]++;
else
res[0]++;
}
return res;
}
}