2024-08-28:用go语言,给定一个从1开始、长度为n的整数数组nums,定义一个函数greaterCount(arr, val)可以返回数组arr中大于val的元素数量。
按照以下规则进行n次操作,将nums中的元素分配到两个数组arr1和arr2中:
1.第一次操作将nums[1]加入arr1。
2.第二次操作将nums[2]加入arr2。
3.对于第i次操作:
3.1.如果arr1中大于nums[i]的元素数量比arr2中大于nums[i]的元素数量多,将nums[i]加入arr1。
3.2.如果arr1中大于nums[i]的元素数量比arr2中大于nums[i]的元素数量少,将nums[i]加入arr2。
3.3.如果arr1和arr2中大于nums[i]的元素数量相等,将nums[i]加入元素数量较少的数组。
3.4.如果仍然相等,则将nums[i]加入arr1。
将arr1和arr2连接起来形成结果数组result。
要求返回整数数组result。
输入:nums = [2,1,3,3]。
输出:[2,3,1,3]。
解释:在前两次操作后,arr1 = [2] ,arr2 = [1] 。
在第 3 次操作中,两个数组中大于 3 的元素数量都是零,并且长度相等,因此,将 nums[3] 追加到 arr1 。
在第 4 次操作中,两个数组中大于 3 的元素数量都是零,但 arr2 的长度较小,因此,将 nums[4] 追加到 arr2 。
在 4 次操作后,arr1 = [2,3] ,arr2 = [1,3] 。
因此,连接形成的数组 result 是 [2,3,1,3] 。
答案2024-08-28:
题目来自leetcode3072。
大体步骤如下:
1.创建一个新的函数greaterCount(arr, val)
,用于计算数组arr
中大于val
的元素数量。
2.定义一个空数组arr1
和arr2
,并创建两个BinaryIndexedTree数据结构tree1
和tree2
。
3.对于数组nums
中的每个元素:
3.1. 将当前元素按照索引排序,并通过Binary Indexed Tree记录每个元素在排序后数组中的位置。
3.2. 进行前两次操作:将nums[0]
加入arr1
,将nums[1]
加入arr2
。
3.3. 从第三个元素开始遍历:
3.3.1.计算arr1
和arr2
中大于当前元素的个数,并根据规则选择将当前元素加入哪个数组,更新对应的Binary Indexed Tree。
4.返回将arr1
和arr2
连接而成的结果数组result
。
总的时间复杂度分析为O(n log n),其中n为数组nums
的长度。
总的额外空间复杂度为O(n),主要是用于存储排序后的数组、索引映射表、两个Binary Indexed Tree结构以及结果数组。