力扣第 387 场周赛第四题 将元素分配到两个数组中 II 二分查找,离散化,线段树

本文主要是介绍力扣第 387 场周赛第四题 将元素分配到两个数组中 II 二分查找,离散化,线段树,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

Problem: 100246. 将元素分配到两个数组中 II

在力扣的题解

赛时没做出来,想了个排序,其实排序总假设最坏的情况即倒序,那肯定超时。当时想到线段树了,但是好久没练没搞出来QAQ,我的板子的线段树下标是从1开始的。
(新题题解力扣审的挺严,我随便写的被截了,所以乖乖算了次复杂度嘤嘤嘤~)

文章目录

  • 思路
  • 解题方法
    • 1.离散化
    • 2.线段树
  • 复杂度
  • Code

思路

我们可以开数组arr1,arr2来统计其中数字出现的数目(下标 i 代表数值,值arr[i]代表出现的次数),然后就可以通过区间和快速得到数组中大于某个数的数目。

因为数据是不断变化的,所以我们得使用树状数组或者线段树。

解题方法

1.离散化

由于数据范围是1~1e9,太大了,不能开1e9的数组。
但是数据量只有1e5,所以我们可以离散化来处理。

离散化后我们要再找这个数的位置,使用二分查找

复制一个数组tmp = nums,我们对tmp进行排序,即完成的离散化。
(由于要使用线段树,所以下标从1开始。)

2.线段树

然后就是通过线段树获取区间和,我们要的是严格大于当前数的数目,找到这个数的下标,下一个位置到数组末尾这段区间的和就是数目。(可能是最后一个数,下一个位置就越界了,特殊处理一下即可)

参考线段树板子:线段树板子

复杂度

时间复杂度:

O ( n l o g 2 n ) O(nlog_2n) O(nlog2n)

二分 l o g 2 n log_2n log2n,建树 n l o g 2 n nlog_2n nlog2n,区间加 l o g 2 n log_2n log2n,区间修改 l o g 2 n log_2n log2n,遍历数组n,各算法间互补干扰,取最大 n l o g 2 n nlog_2n nlog2n

空间复杂度:

O ( 2 n + n ) O(2^n+n) O(2n+n)

对于线段树数组的计算: 1 ∗ 2 ∗ . . . ∗ n 1*2*...*n 12...n,公比为2的等比数列, 1 ∗ 2 n − 1 1 1*\frac{2^{n}-1}{1} 112n1 即 2^n
再加上原数组大小n

Code

类ST是板子;
getmid是二分找下标;
tmp是离散化后下标对应的值的数组。

#define ll long long
class Solution {
public:template<class T>class ST//segment tree{struct node{T val;T t;//懒标记//服务后代node(T v = 0) :val(v), t(0){}};int n = a.size();vector<T>a;vector<node>d;public:void build_tree(int i, int l, int r){if (l == r){d[i].val = a[l];return;}int mid = l + (r - l) / 2;build_tree(i * 2, l, mid);build_tree(i * 2 + 1, mid + 1, r);d[i].val = d[i * 2].val + d[i * 2 + 1].val;}void spread(int i, int l, int r, int aiml, int aimr){int mid = l + (r - l) / 2;if (d[i].t != 0 && l != r){d[i * 2].val += d[i].t * (mid - l + 1);d[i * 2 + 1].val += d[i].t * (r - mid);d[i * 2].t += d[i].t;//可能上上次也没改d[i * 2 + 1].t += d[i].t;d[i].t = 0;}}T getsum(int l, int r){return _getsum(1, 1, n, l, r);}T _getsum(int i, int l, int r, int aiml, int aimr){if (aiml <= l && r <= aimr)//查询区间的子集全部加起来return d[i].val;//访问int mid = l + (r - l) / 2;spread(i, l, r, aiml, aimr);T ret = 0;if (aiml <= mid)ret += _getsum(i * 2, l, mid, aiml, aimr);if (aimr >= mid + 1)ret += _getsum(i * 2 + 1, mid + 1, r, aiml, aimr);return ret;}void update(int l, int r, ll val){_update(1, 1, n, l, r, val);//加并挂标记}void _update(int i, int l, int r, int aiml, int aimr, ll val){if (aiml <= l && r <= aimr){d[i].val += val * (r - l + 1);d[i].t += val;return;}int mid = l + (r - l) / 2;spread(i, l, r, aiml, aimr);if (aiml <= mid)_update(i * 2, l, mid, aiml, aimr, val);if (aimr >= mid + 1)_update(i * 2 + 1, mid + 1, r, aiml, aimr, val);//我们只对叶子更新了,(别多想懒标记)d[i].val = d[i * 2].val + d[i * 2 + 1].val;}ST(vector<T>arr){a = arr;n = a.size() - 1;d = vector<node>((ll)pow((ll)2, (ll)log2(n) + 1 + 1) + 10);build_tree(1, 1, n);}};//vector<int>tmp;int n;int getmid(int aim){int l = 0, r = n;while (l < r){int m = (l + r + 1) / 2;if (tmp[m] <= aim)l = m;else r = m - 1;}return l;}vector<int> resultArray(vector<int>& nums){n = nums.size();//tmp = nums;tmp = vector<int>(n + 1);for (int i = 1; i <= n; i++){tmp[i] = nums[i - 1];}sort(tmp.begin() + 1, tmp.end());vector<int>arr1, arr2, ret;vector<int>sarr1(n+1), sarr2(n+1);ST<int>demo1(sarr1), demo2(sarr2);arr1.push_back(nums[0]);arr2.push_back(nums[1]);int t = getmid(nums[0]);demo1.update(t, t, 1);t = getmid(nums[1]);demo2.update(t, t, 1);for (int i = 2; i < nums.size(); i++){t = getmid(nums[i]);//这个值的下标,我们求右边数目       if (t == n){if (arr1.size() <= arr2.size())arr1.push_back(nums[i]), demo1.update(t, t, 1);elsearr2.push_back(nums[i]), demo2.update(t, t, 1);continue;}int r1 = demo1.getsum(t + 1, n), r2 = demo2.getsum(t + 1, n);if (r1 == r2){if (arr1.size() <= arr2.size())arr1.push_back(nums[i]), demo1.update(t, t, 1);elsearr2.push_back(nums[i]), demo2.update(t, t, 1);}else if (r1 > r2)arr1.push_back(nums[i]), demo1.update(t, t, 1);elsearr2.push_back(nums[i]), demo2.update(t, t, 1);}ret = arr1;for (auto x : arr2)ret.push_back(x);return ret;}
};

这篇关于力扣第 387 场周赛第四题 将元素分配到两个数组中 II 二分查找,离散化,线段树的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/770828

相关文章

Java数组初始化的五种方式

《Java数组初始化的五种方式》数组是Java中最基础且常用的数据结构之一,其初始化方式多样且各具特点,本文详细讲解Java数组初始化的五种方式,分析其适用场景、优劣势对比及注意事项,帮助避免常见陷阱... 目录1. 静态初始化:简洁但固定代码示例核心特点适用场景注意事项2. 动态初始化:灵活但需手动管理代

C++中初始化二维数组的几种常见方法

《C++中初始化二维数组的几种常见方法》本文详细介绍了在C++中初始化二维数组的不同方式,包括静态初始化、循环、全部为零、部分初始化、std::array和std::vector,以及std::vec... 目录1. 静态初始化2. 使用循环初始化3. 全部初始化为零4. 部分初始化5. 使用 std::a

shell编程之函数与数组的使用详解

《shell编程之函数与数组的使用详解》:本文主要介绍shell编程之函数与数组的使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录shell函数函数的用法俩个数求和系统资源监控并报警函数函数变量的作用范围函数的参数递归函数shell数组获取数组的长度读取某下的

如何高效移除C++关联容器中的元素

《如何高效移除C++关联容器中的元素》关联容器和顺序容器有着很大不同,关联容器中的元素是按照关键字来保存和访问的,而顺序容器中的元素是按它们在容器中的位置来顺序保存和访问的,本文介绍了如何高效移除C+... 目录一、简介二、移除给定位置的元素三、移除与特定键值等价的元素四、移除满足特android定条件的元

C语言实现两个变量值交换的三种方式

《C语言实现两个变量值交换的三种方式》两个变量值的交换是编程中最常见的问题之一,以下将介绍三种变量的交换方式,其中第一种方式是最常用也是最实用的,后两种方式一般只在特殊限制下使用,需要的朋友可以参考下... 目录1.使用临时变量(推荐)2.相加和相减的方式(值较大时可能丢失数据)3.按位异或运算1.使用临时

C++从序列容器中删除元素的四种方法

《C++从序列容器中删除元素的四种方法》删除元素的方法在序列容器和关联容器之间是非常不同的,在序列容器中,vector和string是最常用的,但这里也会介绍deque和list以供全面了解,尽管在一... 目录一、简介二、移除给定位置的元素三、移除与某个值相等的元素3.1、序列容器vector、deque

C++常见容器获取头元素的方法大全

《C++常见容器获取头元素的方法大全》在C++编程中,容器是存储和管理数据集合的重要工具,不同的容器提供了不同的接口来访问和操作其中的元素,获取容器的头元素(即第一个元素)是常见的操作之一,本文将详细... 目录一、std::vector二、std::list三、std::deque四、std::forwa

C++原地删除有序数组重复项的N种方法

《C++原地删除有序数组重复项的N种方法》给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度,不要使用额外的数组空间,你必须在原地修改输入数组并在使用O(... 目录一、问题二、问题分析三、算法实现四、问题变体:最多保留两次五、分析和代码实现5.1、问题分析5.

Windows系统下如何查找JDK的安装路径

《Windows系统下如何查找JDK的安装路径》:本文主要介绍Windows系统下如何查找JDK的安装路径,文中介绍了三种方法,分别是通过命令行检查、使用verbose选项查找jre目录、以及查看... 目录一、确认是否安装了JDK二、查找路径三、另外一种方式如果很久之前安装了JDK,或者在别人的电脑上,想

Java中数组转换为列表的两种实现方式(超简单)

《Java中数组转换为列表的两种实现方式(超简单)》本文介绍了在Java中将数组转换为列表的两种常见方法使用Arrays.asList和Java8的StreamAPI,Arrays.asList方法简... 目录1. 使用Java Collections框架(Arrays.asList)1.1 示例代码1.