【竞赛题】6355. 统计公平数对的数目

最新推荐文章于 2024-09-12 19:02:20 发布

程序员小2

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量610

点赞数

分类专栏： leetcode-数组文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/kangbin825/article/details/128992955

版权

leetcode-数组专栏收录该内容

204 篇文章 1 订阅

订阅专栏

给定一个整数数组nums和两个整数lower及upper，程序通过排序和双指针技术计算满足lower<=nums[i]+nums[j]<=upper条件的公平数对的数量。在示例中，展示了如何计算并返回这些数对。

摘要由CSDN通过智能技术生成

题目：

给你一个下标从 0 开始、长度为 n 的整数数组 nums ，和两个整数 lower 和 upper ，返回公平数对的数目。

如果 (i, j) 数对满足以下情况，则认为它是一个公平数对：

0 <= i < j < n，且
lower <= nums[i] + nums[j] <= upper

示例 1：

输入：nums = [0,1,7,4,4,5], lower = 3, upper = 6
输出：6
解释：共计 6 个公平数对：(0,3)、(0,4)、(0,5)、(1,3)、(1,4) 和 (1,5) 。
示例 2：

输入：nums = [1,7,9,2,5], lower = 11, upper = 11
输出：1
解释：只有单个公平数对：(2,3) 。

提示：

1 <= nums.length <= 10^5
nums.length == n
-10^9 <= nums[i] <= 10^9
-10^9 <= lower <= upper <= 10^9

java代码：

class Solution {
    public long countFairPairs(int[] nums, int lower, int upper) {
        Arrays.sort(nums);
        int n = nums.length;

        long c = 0;

        int l = 0;
        int r = n-1;
        while(l<r) {
            int temp = nums[l]+nums[r];
            if(temp<lower) {
                l++;
            }else if(temp>upper) {
                r--;
            }else {
                c++;

                int t2 = r;
                r--;
                while(l<r) {
                    if(nums[r]==nums[r+1]) {
                        c++;
                        r--;
                    }else {
                        temp = nums[l]+nums[r];
                        if(temp<lower) {
                            break;
                        }else {
                            c++;
                            r--;
                        }
                    }
                }

                l++;
                r = t2;
            }
        }

        return c;
    }
}