LeetCode-Python-3011. 判断一个数组是否可以变为有序(模拟）

给你一个下标从 0 开始且全是 正 整数的数组 nums 。

一次 操作 中，如果两个 相邻 元素在二进制下数位为 1 的数目 相同 ，那么你可以将这两个元素交换。你可以执行这个操作 任意次 （也可以 0 次）。

如果你可以使数组变有序，请你返回 true ，否则返回 false 。

示例 1：

输入：nums = [8,4,2,30,15]
输出：true
解释：我们先观察每个元素的二进制表示。 2 ，4 和 8 分别都只有一个数位为 1 ，分别为 "10" ，"100" 和 "1000" 。15 和 30 分别有 4 个数位为 1 ："1111" 和 "11110" 。
我们可以通过 4 个操作使数组有序：
- 交换 nums[0] 和 nums[1] 。8 和 4 分别只有 1 个数位为 1 。数组变为 [4,8,2,30,15] 。
- 交换 nums[1] 和 nums[2] 。8 和 2 分别只有 1 个数位为 1 。数组变为 [4,2,8,30,15] 。
- 交换 nums[0] 和 nums[1] 。4 和 2 分别只有 1 个数位为 1 。数组变为 [2,4,8,30,15] 。
- 交换 nums[3] 和 nums[4] 。30 和 15 分别有 4 个数位为 1 ，数组变为 [2,4,8,15,30] 。
数组变成有序的，所以我们返回 true 。
注意我们还可以通过其他的操作序列使数组变得有序。

示例 2：

输入：nums = [1,2,3,4,5]
输出：true
解释：数组已经是有序的，所以我们返回 true 。

示例 3：

输入：nums = [3,16,8,4,2]
输出：false
解释：无法通过操作使数组变为有序。

提示：

• 1 <= nums.length <= 100
• 1 <= nums[i] <= 2^8

思路：

这道题首先要把 nums 分成二进制 1 的数量相同的子数组，

接着每个子数组内部因为可以任意交换，所以我们只关系最小值和最大值，

接着比较连续的子数组，应该满足当前子数组的最小值大于之前子数组的最大值，这样才能满足总数组有序的题目要求。

时间复杂度：O（NlogK), K = 8

空间复杂度：O（N），理论上可以优化到O（1）一次遍历解决

class Solution:
    def canSortArray(self, nums: List[int]) -> bool:
        pairs = []
        for num in nums:
            pairs.append([num, bin(num).count("1")])
            
        min_val, max_val = inf, -inf
        prev_one_count = None
        intervals = []
        for num, o_count in pairs:
            if prev_one_count is None:
                prev_one_count = o_count
            if prev_one_count == o_count:
                min_val = min(min_val, num)
                max_val = max(max_val, num)
            else:
                # refresh the interval
                prev_one_count = o_count
                intervals.append([min_val, max_val])
                min_val, max_val = num, num
        intervals.append([min_val, max_val])
        for i, interval in enumerate(intervals):
            if i:
                if not (interval[0] >= intervals[i - 1][1]):
                    return False
        return True