474. 一和零
思路:
因为我们的目的是找到含指定0和1的最长串,所以相当于给定背包大小找到能装的最多物品个数。因此这就是一个背包问题!且每个物品只能使用一次!!
我们用dp[i][j][k]表示表示使用j个0和k个1能表示的字符串的最大数量,第 i 个字符要或者不要。
原始版本就是这样~
public int findMaxForm(String[] strs,int m,int n) {
if (strs.length == 0) return 0;
int[][][] dp =new int[strs.length+1][m+1][n+1];
for (int i = 1; i <= strs.length; i++) {
String s = strs[i-1];
int zeros = 0,ones = 0;//统计字符串中一和零的数量。
for(char c:s.toCharArray()) {
if (c == '0') {
zeros++;
}else {
ones++;
}
}
for (int j = 0; j <= m; j++) { //m个0
for (int k = 0; k <= n; k++) { //n个1
//第i个字符要不要 y or n
if (j >= zeros && k >= ones) {
dp[i][j][k] = Math.max(dp[i - 1][j][k], dp[i - 1][j - zeros][k - ones] + 1);
}
else {
dp[i][j][k] = dp[i - 1][j][k];
}
}
}
}
return dp[strs.length][m][n];
}
接下来我们进行空间优化,我们将动态数组的维度降为2,去掉第一个维度,因为第一个维度相当于保存每一层遍历的结果,那么我们干脆直接用上一层的不用特地加一个维度:
public int findMaxForm(String[] strs,int m,int n) {
if (strs.length == 0) return 0;
int[][] dp =new int[m+1][n+1];
for(String s :strs) {
//状态转移方程:dp[i][j] = Math.max(dp[i][j],1+dp[i-numZero][j-numOne])
int zeros = 0,ones = 0;//统计字符串中一和零的数量。
for(char c:s.toCharArray()) {
if (c == '0') {
zeros++;
}else {
ones++;
}
}
for (int i = zeros; i <= m; i++) {
for (int j = ones; j <= n; j++) {
dp[i][j] = Math.max(dp[i][j], 1+dp[i-zeros][j-ones]);
}
}
}
return dp[m][n];
}
你以为这样就可以了吗!!这样就又犯了之前动态规划中背包问题常犯的错!!!
因为在这个背包问题中,每个物品只能用一次,我们如果不单独用一个维度保存上一层的结果的话,一定要在遍历时采取相反的顺序以避免覆盖之前的结果!!!
所以正确代码应该改为:
public int findMaxForm(String[] strs,int m,int n) {
if (strs.length == 0) return 0;
int[][] dp =new int[m+1][n+1];
for(String s :strs) {
//状态转移方程:dp[i][j] = Math.max(dp[i][j],1+dp[i-numZero][j-numOne])
int zeros = 0,ones = 0;//统计字符串中一和零的数量。
for(char c:s.toCharArray()) {
if (c == '0') {
zeros++;
}else {
ones++;
}
}
for (int i = m; i >=zeros; i--) {
for (int j = n; j >= ones; j--) {
dp[i][j] = Math.max(dp[i][j], 1+dp[i-zeros][j-ones]);
}
}
}
return dp[m][n];
}