基础算法（Leetcode）刻意练习第四天——数组

引言

由 “LSGO软件技术团队” 组织的 “基础算法刻意练习” 采用分类别练习的模式，即选择了五个知识点（数组、链表、字符串、树、贪心算法），每个知识点选择了 三个简单、两个中等、一个困难 等级的题目，共计三十道题，利用三十天的时间完成这组刻意练习。以下是我的每日打卡记录：

---

Task04.三数之和

• Leecode第15题

• 难度：中等

• 题目概述：

    给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c 
    使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
    注意：答案中不可以包含重复的三元组。
  
    示例：
    给定数组 nums = [-1, 0, 1, 2, -1, -4]，满足要求的三元组集合为：
    [
    	[-1, 0, 1],
    	[-1, -1, 2]
    ]
  

---

题解思路

---

• 1.HashMap

由 “两数之和” 的 HashMap 联想，只要确定a，在剩下的数中利用 “两数之和” 的方法寻找 b，c即可。
由于禁止重复，可以先将数组排序，方便去重，然后以利用 HashMap 的 “覆盖重复Key” 来防止重复

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result=new ArrayList<>();
        int n=nums.length;
        if(n<3) 
            return result;
        HashMap<Integer,Integer> hm=new HashMap<>();
        Arrays.sort(nums);
        int target;
        for(int i=0;i<n-2;i++){
        	//去除 a 的重复
            if(i>0 && nums[i]==nums[i-1])
                continue;
            //若第一个数大于0，则结束
            if((target=0-nums[i])<0)
                break; 
            //每次开始都要清空HashMap
            hm.clear();
            hm.put(target-nums[i+1],0);//没有使用过则为0
            for(int j=i+2;j<n;j++)
                if(!hm.containsKey(nums[j]))
                    hm.put(target-nums[j],0);
                else if(hm.get(nums[j])==0){
                    result.add(Arrays.asList(nums[i],target-nums[j],nums[j]));
                    hm.put(nums[j],1);//使用过则为1
                }
        }
        return result;
    }
}

提交记录

---

2.双指针（参考题解）

此方法充分利用了升序数组的特点，同样先确定 a 的值，在剩下的数中根据 sum 的大小，利用头尾双指针寻找符合的 b，c
每次 sum==0 都要进行去重（升序数组，相同的数都排在一起）

class Solution {
    public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList();
        int len = nums.length;
        Arrays.sort(nums); // 排序(升序)
        for (int i = 0; i < len ; i++) {
        	// 若第一个数大于0，则结束
            if(nums[i] > 0) break; 
            if(i > 0 && nums[i] == nums[i-1]) continue; // 对a的去重
            int L = i+1;
            int R = len-1;
            while(L < R){
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0){
                    result.add(Arrays.asList(nums[i],nums[L],nums[R]));
                    while (L<R && nums[L] == nums[L+1]) 
                    	L++; // 对b去重
                    while (L<R && nums[R] == nums[R-1]) 
                    	R--; // 对c去重
                    L++;
                    R--;
                }
                else if (sum < 0) L++;
                else if (sum > 0) R--;
            }
        }        
        return result;
    }
}

提交记录