LeetCode 15. 三数之和 回溯+双指针

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。

第一个思路,想到了遍历出所有的三元组,找到符合条件的解即可。

由于不能出现重复的三元组,所以三轮每次循环的值不能重复。

想到回溯,因为可以剪枝嘛!不过好像和优化的三重for循环没有什么区别!!!

//回溯法
class Solution {
    List<List<Integer>> res=new ArrayList<>();

    public List<List<Integer>> threeSum(int[] nums) {
        //对数组进行排序
        Arrays.sort(nums);
        List<Integer> temp=new ArrayList<>();
        dfs(nums,nums.length,-1,0,temp);
        return res;
    }

    private void dfs(int[] nums, int length, int j, int sum,List<Integer> temp) {
        if(sum==0&&temp.size()==3){
            res.add(new ArrayList(temp));
            return;
        }else if(temp.size()>3){
            return;
        }

        for(int i=j+1;i<length;i++){
            if(i==j+1){
                temp.add(nums[i]);
                sum=sum+nums[i];
                dfs(nums,nums.length,i,sum,temp);
                temp.remove(temp.size()-1);
                sum=sum-nums[i];
            }
            else if(nums[i]!=nums[i-1]){
                temp.add(nums[i]);
                sum=sum+nums[i];
                dfs(nums,nums.length,i,sum,temp);
                temp.remove(temp.size()-1);
                sum=sum-nums[i];
            }
        }
    }
}

时间复杂度最坏为O(n的3次方)
提交结果看题解继续改良:

如果我们固定了前两重循环枚举到的元素a,b,那么最后一重循环c是唯一解。那我们向后移动b时,由于数组是排好序的,所以c只可能在上一个c的前面。
这样一个增大,另一个减小的情况,我们可以用双指针来解决,一个指b,一个指c

//双指针
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);

        int left;
        int right;
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < nums.length - 1; i++) {
            if (i == 0 || nums[i] != nums[i - 1]) {
                left = i + 1;
                right = nums.length - 1;
                while (left != right) {
                    if(left>i+1&&nums[left]==nums[left-1]){
                        left++;
                        continue;
                    }
                    if(right<nums.length-1&&nums[right]==nums[right+1]){
                        right--;
                        continue;
                    }
                    if(nums[i]+nums[left]+nums[right]==0){
                        List<Integer> list=new ArrayList<>();
                        list.add(nums[i]);
                        list.add(nums[left]);
                        list.add(nums[right]);
                        res.add(list);
                        left++;
                    }else if(nums[i]+nums[left]+nums[right]>0){
                        right--;
                    }else {
                        left++;
                    }

                }
            }
        }
        return res;
    }
}

时间复杂度为O(n的平方)
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述: 给定一个只包含正整数的非空数组,是否可以将这个数组分成两个子集,使得两个子集的元素和相等。 示例: 输入:[1, 5, 11, 5] 输出:true 解释:数组可以分割成 [1, 5, 5] 和 [11]。 解题思路: 这是一道经典的 0-1 背包问题,可以使用动态规划或者回溯算法解决。 使用回溯算法,需要定义一个 backtrack 函数,该函数有三个参数: - 数组 nums; - 当前处理到的数组下标 index; - 当前已经选择的元素和 leftSum。 回溯过程中,如果 leftSum 等于数组元素和的一半,那么就可以直接返回 true。如果 leftSum 大于数组元素和的一半,那么就可以直接返回 false。如果 index 到达数组末尾,那么就可以直接返回 false。 否则,就对于当前元素,有选择和不选择两种情况。如果选择当前元素,那么 leftSum 就加上当前元素的值,index 就加 1。如果不选择当前元素,那么 leftSum 不变,index 也加 1。最终返回所有可能性的结果中是否有 true。 Java 代码实现: class Solution { public boolean canPartition(int[] nums) { int sum = 0; for (int num : nums) { sum += num; } if (sum % 2 != 0) { return false; } Arrays.sort(nums); return backtrack(nums, nums.length - 1, sum / 2); } private boolean backtrack(int[] nums, int index, int leftSum) { if (leftSum == 0) { return true; } if (leftSum < 0 || index < 0 || leftSum < nums[index]) { return false; } return backtrack(nums, index - 1, leftSum - nums[index]) || backtrack(nums, index - 1, leftSum); } }

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值