[英雄星球六月集训LeetCode解题日报] 第15日树状数组_给定一个数组 nums ,如果 i

本文链接：https://blog.csdn.net/liuliangcan/article/details/125456322

[英雄星球六月集训LeetCode解题日报] 第15日树状数组

日报

第一题不会用树状数组，写个线段树吧。
第二题有点像逆序对。

题目

一、 1157. 子数组中占绝大多数的元素

1157. 子数组中占绝大多数的元素

1. 题目描述

设计一个数据结构，有效地找到给定子数组的多数元素。

子数组的多数元素是在子数组中出现 threshold 次数或次数以上的元素。

实现 MajorityChecker 类:

MajorityChecker(int[] arr) 会用给定的数组 arr 对 MajorityChecker 初始化。
int query(int left, int right, int threshold) 返回子数组中的元素 arr[left…right] 至少出现 threshold 次数，如果不存在这样的元素则返回 -1。

2. 思路分析

这题原先的做法已被hack，是错误的。

正确做法由于难度高，我单开一篇。[LeetCode解题报告] 1157. 子数组中占绝大多数的元素

3. 代码实现

二、 493. 翻转对

493. 翻转对

1. 题目描述

翻转对

难度：困难

给定一个数组 nums ，如果 i < j 且 nums[i] > 2*nums[j] 我们就将 (i, j) 称作一个**重要翻转对**。

你需要返回给定数组中的重要翻转对的数量。

示例 1:

输入: [1,3,2,3,1]
输出: 2

示例 2:

输入: [2,4,3,5,1]
输出: 3

注意:

给定数组的长度不会超过50000。
输入数组中的所有数字都在32位整数的表示范围内。

2. 思路分析

这题有点像逆序对。

我们可以用树状数组tree维护每个数字出现的次数。
倒序遍历，在处理num[i]时，如果tree中存在小于num[i]/2的数据，统计他们出现的个数，就是对num[i]来说组成的翻转对数量。
然后把nums[i]插入树状数组，计数+1。
具体实现时，考虑到奇偶性，对每个num,我们需要统计小于等于(num-1)//2的数出现个数。
这题数据范围未给出，但是给出了数组长度<=5w，因此需要离散化。
需要离散化的数据为nums中所有数据和所有(num-1)//2

3. 代码实现

class BinIndexTree:
    def __init__(self, size):
        self.size = size
        self.bin_tree = [0 for _ in range(size+5)]
    def add(self,i,v):
        while i<=self.size :
            self.bin_tree[i] += v
            i += self.lowbit(i)
    def update(self,i,v):
        val = v - (self.sum(i)-self.sum(i-1))
        self.add(i,val)
    def sum(self,i):
        s = 0
        while i >= 1:
            s += self.bin_tree[i]
            i -= self.lowbit(i)
        return s
    def lowbit(self,x):
        return x&-x

class Solution:
    def reversePairs(self, nums: List[int]) -> int:
        n = len(nums)
        hashed = list(set([(num-1)//2 for num in nums]+nums))
        hashed.sort()
        size = len(hashed)
        tree = BinIndexTree(size)
        ans = 0
        for i in range(n-1,-1,-1):
            num = nums[i]
            pos = bisect_left(hashed,num)+1
            q = bisect_left(hashed,(num-1)//2)+1
            ans += tree.sum(q)

            tree.add(pos,1)
        return ans