leetcode 474 一和零

0-1背包问题,但是有两个背包,两个背包嵌套循环

class Solution {
   public int findMaxForm(String[] strs, int m, int n) {
        //每一个数可以加入也可以不加入
        //0-1背包,先遍历物品,在遍历背包
        // dp[m][n]表示0为m,1为n个时的数组长度
        //获取每一个数组0和1的个数
        int[] nums_0 = new int[strs.length];
        int[] nums_1 = new int[strs.length];
        for(int i = 0; i < strs.length; i++){
            for(int j = 0; j < strs[i].length();j++){
                if (strs[i].charAt(j) == '0')
                    nums_0[i] += 1;
                else
                    nums_1[i] += 1;
            }
        }
        int[][] dp = new int[m + 1][n + 1];
        //先遍历物品,在遍历背包
        // 倒序遍历背包
        for(int i = 0; i < strs.length; i++){
            for(int j = m; j >= nums_0[i];j--){
                for(int k = n; k >= nums_1[i];k--)
                    dp[j][k] = Math.max(dp[j][k],dp[j - nums_0[i]][k - nums_1[i]] + 1);
            }
        }
        return dp[m][n];
    }

}

你可能感兴趣的:(leetcode,算法,动态规划)