[Leetcode] 775. Global and Local Inversions 解题报告

题目

We have some permutation A of [0, 1, ..., N - 1], where N is the length of A.

The number of (global) inversions is the number of i < j with 0 <= i < j < N and A[i] > A[j].

The number of local inversions is the number of i with 0 <= i < N and A[i] > A[i+1].

Return true if and only if the number of global inversions is equal to the number of local inversions.

Example 1:

Input: A = [1,0,2]
Output: true
Explanation: There is 1 global inversion, and 1 local inversion.

Example 2:

Input: A = [1,2,0]
Output: false
Explanation: There are 2 global inversions, and 1 local inversion.

Note:

  • A will be a permutation of [0, 1, ..., A.length - 1].
  • A will have length in range [1, 5000].
  • The time limit for this problem has been reduced.

思路

这道题目的难度实际上在于如何快速地计算global inversions。我们采用的方法有点类似于插入排序,就是逆序地将A中的元素依次加入一个有序集合中,每次加入一个元素后,计算它前面有多少个元素。由于处在它前面的每个元素和它本身在原来的A中都构成了逆序,所以我们累计每次插入一个元素后它前面的元素数量,就可以得到整个A数组中的全局逆序对数。

之后再顺序扫描一遍,得到local inversion,返回他俩是否一样就可以了。算法的空间复杂度是O(n),时间复杂度是O(nlogn),小于暴力法所需要的O(n^2)。

代码

class Solution {
public:
    bool isIdealPermutation(vector<int>& A) {
        set<int> s;
        int size = A.size();
        int global = 0, local = 0;
        for (int i = size - 1; i >= 0; --i) {
            set<int>::iterator it = s.insert(A[i]).first;
            global += distance(s.begin(), it);
        }
        for (int i = 0; i + 1 < size; ++i) {
            local += A[i] > A[i + 1];
        }
        return global == local;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值