两数之和、三数之和

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

注意:答案中不可以包含重复的三元组。

示例 1:

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]

示例 2:

输入:nums = []
输出:[]

示例 3:

输入:nums = [0]
输出:[]

提示:

  • 0 <= nums.length <= 3000
  • -105 <= nums[i] <= 105

题目来源于leetcod,附上漫画更容易理解

算法参考博客:漫画算法题:两数之和与三数之和_程序员小灰的博客-CSDN博客

两数之和算法思路:以第i个数为第一个数,这时候只要找第元素是trage - arr[i]的就行了,i后面的元素用哈希表存储,查找的时间复杂度就能降低到1.

    public static int[] addTwoNumbers(int[] arr, int trage){
        Map<Integer, Integer> map = new HashMap<>();
        int length = arr.length;
        // map<元素,下标>
        for(int i = 1; i < length; i++){
            map.put(arr[i], i);
        }

        for(int i = 0; i < length; i++){
            int temp = trage - arr[i];
            if( map.containsValue(temp) && map.get(temp) != i){
                return new int[]{i, map.get(temp)};
            }
        }

        return null;
    }

三数之和算法思路:先把数据排序。也是以第i个元素为第1个元素,则寻找i之后的元素加起来是trage - arr[i]就行了,使用双指针指向i+1和length-1下标的元素,相加大于就k--,小于就j++,直到j j == k。因为leetcode要求不能有重复元素,所以在最后的return用stream()做去重。

public static List<List<Integer>> addThreeNumbers(int[] arr, int trage){
        arr = Arrays.stream(arr).sorted().toArray();
        int length = arr.length;
        List<List<Integer>> resList = new ArrayList<>();

        for(int i = 0; i < length; i++){
            int temp = trage - arr[i];
            int j = i + 1;
            int k = length - 1;
            for(; j < k;){
                if(temp > (arr[j] + arr[k])){
                    j ++;
                    continue;
                }
                if(temp < (arr[j] + arr[k])){
                    k --;
                    continue;
                }
                if(temp == (arr[j] + arr[k])){
                    resList.add(Arrays.asList(arr[i], arr[j], arr[k]));
                    j ++;
                    k --;
                    continue;
                }
            }
        }

        return resList.stream().distinct().collect(Collectors.toList());
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值