力扣第 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

相关文章

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.

java两个List的交集,并集方式

《java两个List的交集,并集方式》文章主要介绍了Java中两个List的交集和并集的处理方法,推荐使用Apache的CollectionUtils工具类,因为它简单且不会改变原有集合,同时,文章... 目录Java两个List的交集,并集方法一方法二方法三总结java两个List的交集,并集方法一

nginx upstream六种方式分配小结

《nginxupstream六种方式分配小结》本文主要介绍了nginxupstream六种方式分配小结,包括轮询、加权轮询、IP哈希、公平轮询、URL哈希和备份服务器,具有一定的参考价格,感兴趣的可... 目录1 轮询(默认)2 weight3 ip_hash4 fair(第三方)5 url_hash(第三

C++一个数组赋值给另一个数组方式

《C++一个数组赋值给另一个数组方式》文章介绍了三种在C++中将一个数组赋值给另一个数组的方法:使用循环逐个元素赋值、使用标准库函数std::copy或std::memcpy以及使用标准库容器,每种方... 目录C++一个数组赋值给另一个数组循环遍历赋值使用标准库中的函数 std::copy 或 std::

使用C++实现链表元素的反转

《使用C++实现链表元素的反转》反转链表是链表操作中一个经典的问题,也是面试中常见的考题,本文将从思路到实现一步步地讲解如何实现链表的反转,帮助初学者理解这一操作,我们将使用C++代码演示具体实现,同... 目录问题定义思路分析代码实现带头节点的链表代码讲解其他实现方式时间和空间复杂度分析总结问题定义给定