leetcode 1. 两数之和 - 简单 & leetcode 15. 三数之和 - 难度中等

1.1 题目描述

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1]

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

提示：

2 <= nums.length <= 103
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

1.2 题解

思路：借助map，遍历数组，对于当前元素，如果在map中能找到 target-当前元素，则返回这二者的下标；否则，将当前元素加入map中。

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[]}
 */
var twoSum = function(nums, target) {
    const map = new Map();
    for(let i = 0; i < nums.length; i++){
        if(map.has(target - nums[i])){
            return [map.get(target - nums[i]), i];
        }else{
            map.set(nums[i], i);
        }
    }
};

时间复杂度：O(n)，
空间复杂度：O(n)。

2.1 题目描述

给你一个包含 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.2 题解

思路：
首先对数组排序，遍历数组，i 指向当前遍历元素，left是它左边的元素，right是从数组右边开始遍历的元素；

• 如果nums[i]>0，说明三者之和必然大于0，直接跳出本轮循环；
• 当三者之和等于0时，更新left和right, 这里注意去重；
• 三者之和小于0，left 右移；
• 三者之和大于0, right 左移。

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    //直接排除
    if(nums === null || nums.length < 3){
        return [];
    }
    const res =[];
    //对数组排序,需要自己定义排序规则
    nums.sort((a,b) => a-b);
    //双指针
    for(let i = 0; i < nums.length; i++){
        if(nums[i] > 0){
            break;
        }
        //去重
        if(i > 0 && nums[i] === nums[i-1]){
            continue;
        }
        let left = i + 1;
        let right = nums.length - 1;
        while(left < right){
            if(nums[i] + nums[left] + nums[right] === 0){
                res.push([nums[i], nums[left], nums[right]]);
                //去重
                while(left < right && nums[left] === nums[left+1]){
                    left++;
                }
                //去重
                while(left < right && nums[right] === nums[right-1]){
                    right--;
                }
                left++;
                right--;
            }else if(nums[i] + nums[left] + nums[right] < 0){
                left++;
            }else{
                right--;
            }
        }
    }
    return res;
};

时间复杂度：O(n*n)，
空间复杂度：O(logn)，不计返回结果开辟的空间，这里给数组排序需要logn的空间。

注意：
数组的sort()：就地排序，默认是将数组元素转换为字符串后比较它们的UTF-16的值。
所以在题目中，对数组进行排序，需要定义排序规则，即 nums.sort((a, b) => a-b);