力扣算法学习day29-1

力扣算法学习day29-1

474-一和零

题目

代码实现

class Solution {
    public int findMaxForm(String[] strs, int m, int n) {
        // dp 直接想到并解决 速度 18ms
        // 题目分析：子集最多有m个0，n个1，且每个元素只能取一次，很像01背包了，但是有两个未知量。
        // 如果单独m个0，或n个1的话，就是标准的01背包问题，m或n为背包重量，元素中含有的0或1就为
        // 每个元素的重量，每个元素的价值就是1，相当于求放入最多元素，也就是价值最高。
        // 思路：这里m和n我最直接想到的就是把背包想象成一个立体的背包，简单的说就是背包容量有两个维度
        // m，n都需要考虑，对二维的背包时，竖坐标是元素，横坐标是重量，那么可以想象成竖坐标依然是元素
        // 但是横坐标变成一个面，m，n都需要考虑。
        // 1.创建dp数组
        int[][] dp = new int[m+1][n+1];

        // 2.确定迭代公式：dp[i][j] = Math.max(dp[i][j],dp[i-x][j-y]+1).x为元素的0的个数，y为元素1的个数。
        // 三维其实就是 dp[z][i][j] = Math.max(dp[z-1][i][j],dp[z-1][i-x][j-y]+1),再由三维压缩成二维即可。
        // 3.初始化,初始化为0即可。
        // 4.确定遍历顺序，必须先元素。然后才是m，n。
        for(int i = 0;i < strs.length;i++){
            // 需要统计该元素有是0和1的个数。
            char[] str = strs[i].toCharArray();
            int zero = 0;
            int one = 0;
            for(int j = 0;j < str.length;j++){
                if(str[j] == '0'){
                    zero++;
                } else if(str[j] == '1'){
                    one++;
                }
            }

            for(int j = m;j >= zero;j--){
                for(int z = n;z >= one;z--){
                    dp[j][z] = Math.max(dp[j][z],dp[j-zero][z-one]+1);
                }
            }
        }

        return dp[m][n];
    }
}