Leetcode416:分割等和子集&&Leetcode1305:两棵二叉搜索树中的所有元素

Leetcode416:分割等和子集

• 题目：给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。

• 思路：0-1背包问题

• 代码如下：

class Solution {
    public boolean canPartition(int[] nums) {
        //长度为0或者空数组，不可能分割等和子集
        if (nums.length==0||nums==null){
            return false;
        }
        int sum = 0;
        for (int num : nums){
            sum += num;
        }
        int target = sum/2;
        //总和为奇数，不可能分割等和子集
        if (sum%2!=0){
            return false;
        }
      	//定义一维数组，dp[i]表示 背包总容量是i，最大可以凑成j的子集总和为dp[i]
        int[] dp = new int[target+1];
        for (int i = 0;i< nums.length;i++){
            for(int j = target;j>=nums[i];j--){
                dp[j] = Math.max(dp[j-nums[i]]+nums[i],dp[j]);
            }
        }
    
        return dp[target]==target;

    }
}

Leetcode1305:两棵二叉搜索树中的所有元素

• 题目：给你 root1 和 root2 这两棵二叉搜索树。请你返回一个列表，其中包含 两棵树 中的所有整数并按 升序 排序。
• 思路：遍历两棵二叉树（中序遍历排序），然后收集到两个数组l1和l2，最后将两个数组l1和l2进行归并排序成结果数组res。
• 代码如下：

class Solution {
    //存放比较时的最大值（为归并排序准备的最大值）
    int INF = 0x3f3f3f3f;

    public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
        //存放最终结果数组
        List<Integer> res = new ArrayList<>();
        //存放完成升序排序的root1 线性数组
        List<Integer> l1 = new ArrayList<>();
        //存放完成升序排序的root2 线性数组
        List<Integer> l2 = new ArrayList<>();
        //将二叉树树型结构转换为线性结构
        dfs(root1,l1);
        dfs(root2,l2);
        int i = 0, j = 0, n = l1.size(), m = l2.size();
        //对数组l1和l2进行升序排序（归并排序）
        while (i < n || j < m){
            int a = i < n ? l1.get(i) : INF;
            int b = j < m ? l2.get(j) : INF;
            if(a < b){
                res.add(a);
                i++;
            }else{
                res.add(b);
                j++;
            }
        }
        return res;
    }
    //中序遍历排序
    void dfs(TreeNode root,List<Integer> list){
        if(root == null){
            return ;
        }
        dfs(root.left,list);
        list.add(root.val);
        dfs(root.right,list);
    }
}