README.md

comments	difficulty	edit_url	tags
true	中等	https://github.com/doocs/leetcode/edit/main/solution/0200-0299/0259.3Sum%20Smaller/README.md	数组 双指针 二分查找 排序

259. 较小的三数之和 🔒

English Version

题目描述

给定一个长度为 n 的整数数组和一个目标值 target ，寻找能够使条件 nums[i] + nums[j] + nums[k] < target 成立的三元组  i, j, k 个数（0 <= i < j < k < n）。

 

示例 1：

输入: nums = [-2,0,1,3], target = 2
输出: 2 
解释: 因为一共有两个三元组满足累加和小于 2:
     [-2,0,1]
     [-2,0,3]

示例 2：

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

示例 3：

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

 

提示:

• n == nums.length
• 0 <= n <= 3500
• -100 <= nums[i] <= 100
• -100 <= target <= 100

解法

方法一：排序 + 双指针 + 枚举

由于元素的顺序不影响结果，我们可以先对数组进行排序，然后使用双指针的方法来解决这个问题。

我们先将数组排序，然后枚举第一个元素 $\mathit{\text{nums}}[i]$，并在 $\mathit{\text{nums}}[i+1:n-1]$ 的区间内使用双指针分别指向 $\mathit{\text{nums}}[j]$ 和 $\mathit{\text{nums}}[k]$，其中 $j$ 是 $\mathit{\text{nums}}[i]$ 的下一个元素，而 $k$ 是数组的最后一个元素。

• 如果 $\mathit{\text{nums}}[i]+\mathit{\text{nums}}[j]+\mathit{\text{nums}}[k]<\mathit{\text{target}}$，那么对于任意 $j<k^{\prime }\le k$ 的元素，都有 $\mathit{\text{nums}}[i]+\mathit{\text{nums}}[j]+\mathit{\text{nums}}[k^{\prime }]<\mathit{\text{target}}$，一共有 $k-j$ 个这样的 $k^{\prime }$，我们将 $k-j$ 累加到答案中。接下来，将 $j$ 右移一个位置，继续寻找下一个满足条件的 $k$，直到 $j\ge k$ 为止。
• 如果 $\mathit{\text{nums}}[i]+\mathit{\text{nums}}[j]+\mathit{\text{nums}}[k]\ge \mathit{\text{target}}$，那么对于任意 $j\le j^{\prime }<k$ 的元素，都不可能使得 $\mathit{\text{nums}}[i]+\mathit{\text{nums}}[j^{\prime }]+\mathit{\text{nums}}[k]<\mathit{\text{target}}$，因此我们将 $k$ 左移一个位置，继续寻找下一个满足条件的 $k$，直到 $j\ge k$ 为止。

枚举完所有的 $i$ 后，我们就得到了满足条件的三元组的个数。

时间复杂度 $O(n^2)$，空间复杂度 $O(\log n)$。其中 $n$ 是数组 $\mathit{\text{nums}}$ 的长度。

Python3

class Solution:
    def threeSumSmaller(self, nums: List[int], target: int) -> int:
        nums.sort()
        ans, n = 0, len(nums)
        for i in range(n - 2):
            j, k = i + 1, n - 1
            while j < k:
                x = nums[i] + nums[j] + nums[k]
                if x < target:
                    ans += k - j
                    j += 1
                else:
                    k -= 1
        return ans

Java

class Solution {
    public int threeSumSmaller(int[] nums, int target) {
        Arrays.sort(nums);
        int ans = 0, n = nums.length;
        for (int i = 0; i + 2 < n; ++i) {
            int j = i + 1, k = n - 1;
            while (j < k) {
                int x = nums[i] + nums[j] + nums[k];
                if (x < target) {
                    ans += k - j;
                    ++j;
                } else {
                    --k;
                }
            }
        }
        return ans;
    }
}

C++

class Solution {
public:
    int threeSumSmaller(vector<int>& nums, int target) {
        ranges::sort(nums);
        int ans = 0, n = nums.size();
        for (int i = 0; i + 2 < n; ++i) {
            int j = i + 1, k = n - 1;
            while (j < k) {
                int x = nums[i] + nums[j] + nums[k];
                if (x < target) {
                    ans += k - j;
                    ++j;
                } else {
                    --k;
                }
            }
        }
        return ans;
    }
};

Go

func threeSumSmaller(nums []int, target int) (ans int) {
	sort.Ints(nums)
	n := len(nums)
	for i := 0; i < n-2; i++ {
		j, k := i+1, n-1
		for j < k {
			x := nums[i] + nums[j] + nums[k]
			if x < target {
				ans += k - j
				j++
			} else {
				k--
			}
		}
	}
	return
}

TypeScript

function threeSumSmaller(nums: number[], target: number): number {
    nums.sort((a, b) => a - b);
    const n = nums.length;
    let ans = 0;
    for (let i = 0; i < n - 2; ++i) {
        let [j, k] = [i + 1, n - 1];
        while (j < k) {
            const x = nums[i] + nums[j] + nums[k];
            if (x < target) {
                ans += k - j;
                ++j;
            } else {
                --k;
            }
        }
    }
    return ans;
}

JavaScript

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number}
 */
var threeSumSmaller = function (nums, target) {
    nums.sort((a, b) => a - b);
    const n = nums.length;
    let ans = 0;
    for (let i = 0; i < n - 2; ++i) {
        let [j, k] = [i + 1, n - 1];
        while (j < k) {
            const x = nums[i] + nums[j] + nums[k];
            if (x < target) {
                ans += k - j;
                ++j;
            } else {
                --k;
            }
        }
    }
    return ans;
};