2164. 对奇偶下标分别排序

2022-02-09  本文已影响0人  Abeants

给你一个下标从 0 开始的整数数组 nums 。根据下述规则重排 nums 中的值:

按 非递增 顺序排列 nums 奇数下标 上的所有值。
举个例子,如果排序前 nums = [4,1,2,3] ,对奇数下标的值排序后变为 [4,3,2,1] 。奇数下标 1 和 3 的值按照非递增顺序重排。
按 非递减 顺序排列 nums 偶数下标 上的所有值。
举个例子,如果排序前 nums = [4,1,2,3] ,对偶数下标的值排序后变为 [2,1,4,3] 。偶数下标 0 和 2 的值按照非递减顺序重排。
返回重排 nums 的值之后形成的数组。

示例 1:
输入:nums = [4,1,2,3]
输出:[2,3,4,1]
解释:
首先,按非递增顺序重排奇数下标(1 和 3)的值。
所以,nums 从 [4,1,2,3] 变为 [4,3,2,1] 。
然后,按非递减顺序重排偶数下标(0 和 2)的值。
所以,nums 从 [4,1,2,3] 变为 [2,3,4,1] 。
因此,重排之后形成的数组是 [2,3,4,1] 。

示例 2:
输入:nums = [2,1]
输出:[2,1]
解释:
由于只有一个奇数下标和一个偶数下标,所以不会发生重排。
形成的结果数组是 [2,1] ,和初始数组一样。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sort-even-and-odd-indices-independently
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路及方法

优先队列,将偶数下标存入小根堆,奇数下标存入大根堆,然后按序输出。

class Solution {
    public int[] sortEvenOdd(int[] nums) {
        int len = nums.length;
        
        PriorityQueue<Integer> odd = new PriorityQueue<>((a, b) -> b -a);
        PriorityQueue<Integer> even = new PriorityQueue<>();

        // 入队
        for (int i = 0; i < len; i++) {
            if (i % 2 == 1) {
                odd.offer(nums[i]);
            } else {
                even.offer(nums[i]);
            }
        }
        
        for (int i = 0; i < len; i++) {
            if (i % 2 == 1) {
                nums[i] = odd.poll();
            } else {
                nums[i] = even.poll();
            }
        }
        
        return nums;
    }
}

结果如下:

上一篇 下一篇

猜你喜欢

热点阅读