力扣493.翻转对

力扣493.翻转对

• 贪心 + 树状数组

  • 将2*v和v一起加入st集合离散化
  • 最小的数编号越小
  • 之后在把nums[i]加入树状数组时，求编号低的即可

• 为什么要把2*v也一起离散化？

  • 因为我们通过比较编号来比较数字大小
  • 如果2*v不离散化，它就没有对应编号，无法比较

•   const int N = 100010;
    int tr[N],n,limit;
    class Solution {
    public:
        int reversePairs(vector<int>& nums) {
            n = nums.size();
            set<long long> st;
            for(int v:nums)
            {
                st.insert(v);
                st.insert((long long)2*v);
            }
            int i=1;
            unordered_map<long long,int> mp;
            for(auto s:st)
                mp[s] = i++;
            memset(tr,0,sizeof tr);
            int count=0;
            limit = st.size();
            for(int i=n-1;i>=0;i--)
            {
                int r = mp[nums[i]];
                count += sum(r-1);
                add(mp[(long long)nums[i]*2],1);
            }
            return count;
        }
        int lowbit(int x)
        {
            return x & -x;
        }
        void add(int x,int c)
        {
            for(int i=x;i<=limit;i+=lowbit(i)) tr[i] += c;
        }
        int sum(int x)
        {
            int res=0;
            for(int i=x;i;i-=lowbit(i)) res += tr[i];
            return res;
        }
    };