题目:
给你一个下标从 0 开始的整数数组 nums
。
一开始,所有下标都没有被标记。你可以执行以下操作任意次:
- 选择两个 互不相同且未标记 的下标
i
和j
,满足2 * nums[i] <= nums[j]
,标记下标i
和j
。
请你执行上述操作任意次,返回 nums
中最多可以标记的下标数目。
示例 1:
输入:nums = [3,5,2,4]
输出:2
解释:第一次操作中,选择 i = 2 和 j = 1 ,操作可以执行的原因是 2 * nums[2] <= nums[1] ,标记下标 2 和 1 。
没有其他更多可执行的操作,所以答案为 2 。
示例 2:
输入:nums = [9,2,5,4]
输出:4
解释:第一次操作中,选择 i = 3 和 j = 0 ,操作可以执行的原因是 2 * nums[3] <= nums[0] ,标记下标 3 和 0 。
第二次操作中,选择 i = 1 和 j = 2 ,操作可以执行的原因是 2 * nums[1] <= nums[2] ,标记下标 1 和 2 。
没有其他更多可执行的操作,所以答案为 4 。
示例 3:
输入:nums = [7,6,8]
输出:0
解释:没有任何可以执行的操作,所以答案为 0 。
提示:
1 <= nums.length <= 105
1 <= nums[i] <= 109
思路:
由于长度为 n 的数组最多只会产生 ⌊2 / n⌋ 对匹配,因此对数组从小到大排序以后,我们将数组一分为二,左侧元素只会与右侧元素匹配。
具体的,我们令 m=⌊2 / n⌋,尝试将下标在 [0, m − 1] 范围内的元素 nums[i] 与下标在 [m, n − 1] 范围内的元素 nums[j] 进行匹配。我们从小到大枚举 i,然后找到最小的 j 使其满足 2 * nums[i] ≤ nums[j]。那些未满足条件而被跳过的 nums[j] 将被忽略。持续这一过程,直到 i = m 或 j = n。
代码:
class Solution {
public:
int maxNumOfMarkedIndices(vector<int>& nums) {
sort(nums.begin(), nums.end());
int n = nums.size();
int m = n / 2;
int res = 0;
for (int i = 0, j = m; i < m && j < n; i++) {
while (j < n && 2 * nums[i] > nums[j]) {
j++;
}
if (j < n) {
res += 2;
j++;
}
}
return res;
}
};