【力扣】3134.找出唯一性数组的中位数

题目描述

给你一个整数数组 nums 。数组 nums 的 唯一性数组 是一个按元素从小到大排序的数组，包含了 nums 的所有非空子数组中不同元素的个数。

换句话说，这是由所有 0 <= i <= j < nums.length 的 distinct(nums[i..j]) 组成的递增数组。

其中，distinct(nums[i..j]) 表示从下标 i 到下标 j 的子数组中不同元素的数量。

返回 nums 唯一性数组 的 中位数 。

注意，数组的 中位数 定义为有序数组的中间元素。如果有两个中间元素，则取值较小的那个。

示例 1：

输入：nums = [1,2,3]

输出：1

解释：

nums 的唯一性数组为 [distinct(nums[0..0]), distinct(nums[1..1]), distinct(nums[2..2]), distinct(nums[0..1]), distinct(nums[1..2]), distinct(nums[0..2])]，即 [1, 1, 1, 2, 2, 3] 。唯一性数组的中位数为 1 ，因此答案是 1 。

示例 2：

输入：nums = [3,4,3,4,5]

输出：2

解释：

nums 的唯一性数组为 [1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3] 。唯一性数组的中位数为 2 ，因此答案是 2 。

示例 3：

输入：nums = [4,3,5,4]

输出：2

解释：

nums 的唯一性数组为 [1, 1, 1, 1, 2, 2, 2, 3, 3, 3] 。唯一性数组的中位数为 2 ，因此答案是 2 。

提示：

• 1 <= nums.length <= 105
• 1 <= nums[i] <= 105

AC代码

int hashh[100005];

class Solution {
public:
    /* 检测数组中不同元素数目小于等于 t 的连续子数组数目是否大于等于 median */
    bool search(vector<int>& nums, int mid, long long median) {
        memset(hashh, 0, sizeof(hashh));
        int n = nums.size();
        int total = 0;
        int left = 0;
        long long ans = 0;
        //滑动窗口
        for (int right = 0; right < n; right++) {
            hashh[nums[right]]++;
            if (hashh[nums[right]] == 1)
                total++;
            while (total > mid) {
                hashh[nums[left]]--;
                if (hashh[nums[left]] == 0) total--;
                left++;
            }
            ans += right - left + 1;
        }
        return ans <= median;
    }

    int medianOfUniquenessArray(vector<int>& nums) {
        int n = nums.size();
        long long median = ((long long)n * (n + 1) / 2 + 1) / 2 - 1;
        int l = 0, r = n;
        while (l <= r) {
            int mid = (l + r) /2 ;
            if (search(nums, mid, median)) {
                l = mid + 1;
            } 
            else r = mid - 1;
        }
        return l;
    }
};