Leetcode.2659 将数组清空

题目链接

Leetcode.2659 将数组清空 Rating ： 2282

题目描述

给你一个包含若干 互不相同 整数的数组 $nums$ ，你需要执行以下操作 直到数组为空 ：

• 如果数组中第一个元素是当前数组中的 最小值 ，则删除它。
• 否则，将第一个元素移动到数组的 末尾 。

请你返回需要多少个操作使 $nums$ 为空。

示例 1：

输入：nums = [3,4,-1]
输出：5

示例 2：

输入：nums = [1,2,4,3]
输出：5

示例 3：

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

提示：

• $1<=nums.length<=10^5$
• $-10^9<=nums[i]<=10^9$
• $nums$ 中的元素 互不相同 。

解法：树状数组 & 排序

• 如果数组中第一个元素是当前数组中的 最小值 ，则删除它。
• 否则，将第一个元素移动到数组的 末尾 。

对于这样的操作，我们可以换一种角度：我们将原数组 $nums$ 看成是 循环数组，移动元素到末尾看成一个指针在数组里移动，删除最小值 看成是 标记最小值。最后返回的是标记完 $nums$ 数组的操作数。

由于我们每次标记的都是剩下的最小值，所以我们需要对 $nums$ 按 从小到大 的顺序排序。由于我们也要获取 $nums$ 中元素的下标信息，所以我们直接对 { n u m s [ i ] , i + 1 } \{ nums[i] , i + 1 \} {nums[i],i+1} 这样的二元组序列排序（按 $nums[i]$ 从小到大排序）。

假设最小值的下标是 $i$ , 次小值的下标是 $j$。我们标记完 $i$ 之后，在标记 $j$ 的操作数量就是 $j-i-区间[i,j]中已经被标记的位置数量$。

我们要快速的得出 一个区间内的某一元素数量 ，可以使用 : 树状数组

时间复杂度： $O(n\times logn)$

C++代码：

using LL = long long;
using PII = pair<int,int>;

class Solution {
public:
    long long countOperationsToEmptyArray(vector<int>& nums) {
        int n = nums.size();

        int tree[n + 1];
        memset(tree,0,sizeof tree);

        auto lb = [](int x){
            return x & (-x);
        };

        //给树状数组的 pos 位置 + 1
        auto add = [&](int pos){
            for(;pos <= n;pos += lb(pos)) tree[pos]++;
        };

        //查询 前pos位置个前缀和
        auto query = [&](int pos){
            LL sum = 0;
            for(;pos;pos -= lb(pos)) sum += tree[pos];
            return sum;
        };

        vector<PII> a;
        for(int i = 0;i < n;i++) a.emplace_back(nums[i],i + 1);
        sort(a.begin() , a.end());

        LL ans = 0;
        int cur = 0;

        for(auto &p:a){
            //要标记的位置在 cur 指针前面，说明需要把指针 cur 先移到末尾，再移动到前面
            if(p.second < cur){
                ans += n - cur;

                //再去掉 (cur,n] 已经被标记的位置数量
                ans -= query(n) - query(cur);

                //现在指针重新指向数组开头
                cur = 0; 
            }

            //把指针 cur 移动到要标记的位置 p.second
            ans += p.second - cur;

            //去除掉 (cur,p.second] 中已经被标记的位置数量
            ans -= query(p.second) - query(cur);

            //标记 p.second
            add(p.second);

            //指针 cur 此时就指向这个位置
            cur = p.second;
        }
        return ans;
    }
};