compute exact number of triples of distinct elements in large array

最新推荐文章于 2023-08-04 14:41:24 发布

Seth_Lin

最新推荐文章于 2023-08-04 14:41:24 发布

阅读量495

点赞数

分类专栏： Google面试

Google面试专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种高效算法，用于找出数组中所有不同元素构成的三元组，使得这三个数的和小于或等于给定的目标值。通过排序和双指针技巧，实现了O(n^2)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array of N=10^6 int32 and an int32 X, compute the exact number of triples (a, b, c) of distinct elements of the array so that a + b + c <= X

    public int threeSumSmaller(int[] nums, int target) {
        if (nums.length < 3) {
            return 0;
        }
        Arrays.sort(nums);
        int res = 0;
        for (int i = 0; i < nums.length - 2; ++i) {
            if (nums[i] + nums[i + 1] + nums[i + 2] >= target) {
                break;
            }
            int curr = nums[i];
            int lo = i + 1, hi = nums.length - 1;
            while (lo < hi) {
                while (lo < hi && curr + nums[lo] + nums[hi] >= target) {
                    --hi;
                }
                res += hi - lo;
                ++lo;
            }
        }
        return res;
    }