nSum的java解法

两数之和

1. 两数之和 - 力扣(LeetCode)

可以使用哈希表的方法来解决

  1. 如果哈希表中不存在 target - a 这个元素,将这个元素值座位 key,下标作为 value 加入到哈希表中
  2. 如果存在,返回两个元素的下标即可
  3. 如果遍历到数组末尾都没有找到,返回空数组
class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map map = new HashMap();
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(target - nums[i])) {
                return new int[]{map.get(target - nums[i]),i};
            } else {
                map.put(nums[i],i);
            }
        }

        return new int[0];
    }
}

三数之和

15. 三数之和 - 力扣(LeetCode)

可以用排序加双指针的做法

先整体排序,然后一前一后两个指针进行遍历,方法和二分查找类似

class Solution {
    public List> threeSum(int[] nums) {
        List> ans = new ArrayList();
        int len = nums.length;
        if (len < 3) {return ans;}
        Arrays.sort(nums);

        for (int i = 0; i < len; i++) {
            if (nums[i] > 0) {break;}
            if (i > 0 && nums[i] == nums[i-1]) {continue;}
            int left = i + 1, right = len - 1;

            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];

                if (sum == 0) {
                    ans.add(Arrays.asList(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 (sum < 0) {
                    left++;
                } else if (sum > 0) {
                    right--;
                }
            }
        }
        return ans;
    }
}

1. 初始化:创建一个空列表 `ans` 用于存放符合条件的三元组。
    
2. 数组排序:对输入的数组 `nums` 进行升序排序。
    
3. 三重循环:使用三重循环,外层循环遍历数组 `nums` 的元素,固定一个元素 `nums[i]`,内层双指针循环找到和为零的三元组。
    
4. 避免重复解:在外层循环中,如果当前元素 `nums[i]` 大于 0,则直接退出循环,因为排序后后续元素不可能使得三元组和为零。如果 `nums[i]` 与前一个元素相同,则跳过当前循环,避免重复解。
    
5. 双指针法:使用两个指针 `left` 和 `right`,分别指向当前元素的下一个位置和数组末尾。在内层循环中,计算当前三元组的和 `sum = nums[i] + nums[left] + nums[right]`。
    
6. 判断和的大小:如果 `sum == 0`,则将三元组 `[nums[i], nums[left], nums[right]]` 添加到结果列表中,并进行去重操作。同时,移动指针 `left` 向右移动到下一个不同的元素,`right` 向左移动到下一个不同的元素。如果 `sum < 0`,说明当前和偏小,需要增大,因此移动 `left` 指针;如果 `sum > 0`,说明当前和偏大,需要减小,因此移动 `right` 指针。
    
7. 返回结果:循环结束后,返回存放结果的列表 `ans`。

你可能感兴趣的:(java,leetcode,算法)