LeetCode 1825. 求出 MK 平均值（set + queue）

文章目录

1. 题目
2. 解题

1. 题目

给你两个整数 m 和 k ，以及数据流形式的若干整数。
你需要实现一个数据结构，计算这个数据流的 MK 平均值。

MK 平均值按照如下步骤计算：

如果数据流中的整数少于 m 个，MK 平均值为 -1 ，否则将数据流中最后 m 个元素拷贝到一个独立的容器中。
从这个容器中删除最小的 k 个数和最大的 k 个数。
计算剩余元素的平均值，并向下取整到最近的整数。

请你实现 MKAverage 类：

MKAverage(int m, int k) 用一个空的数据流和两个整数 m 和 k 初始化 MKAverage 对象。
void addElement(int num) 往数据流中插入一个新的元素 num 。
int calculateMKAverage() 对当前的数据流计算并返回 MK 平均数，结果需向下取整到最近的整数。

示例 1：
输入：
["MKAverage", "addElement", "addElement", "calculateMKAverage", "addElement", "calculateMKAverage", "addElement", "addElement", "addElement", "calculateMKAverage"]
[[3, 1], [3], [1], [], [10], [], [5], [5], [5], []]
输出：
[null, null, null, -1, null, 3, null, null, null, 5]
解释：
MKAverage obj = new MKAverage(3, 1);
obj.addElement(3);        // 当前元素为 [3]
obj.addElement(1);        // 当前元素为 [3,1]
obj.calculateMKAverage(); // 返回 -1 ，因为 m = 3 ，但数据流中只有 2 个元素
obj.addElement(10);       // 当前元素为 [3,1,10]
obj.calculateMKAverage(); // 最后 3 个元素为 [3,1,10]// 删除最小以及最大的 1 个元素后，容器为 [3]// [3] 的平均值等于 3/1 = 3 ，故返回 3
obj.addElement(5);        // 当前元素为 [3,1,10,5]
obj.addElement(5);        // 当前元素为 [3,1,10,5,5]
obj.addElement(5);        // 当前元素为 [3,1,10,5,5,5]
obj.calculateMKAverage(); // 最后 3 个元素为 [5,5,5]// 删除最小以及最大的 1 个元素后，容器为 [5]// [5] 的平均值等于 5/1 = 5 ，故返回 5提示：
3 <= m <= 10^5
1 <= k*2 < m
1 <= num <= 10^5
addElement 与 calculateMKAverage 总操作次数不超过 10^5 次。

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/finding-mk-average
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

使用 3个 multiset 记录3段数据，同时queue记录数据，更新删除，只保留 m 个
记录中间段的 sum

class MKAverage {int m, k;long long sum = 0;//中间的数的和queue<int> q;multiset<int> big, mid;multiset<int, greater<int>> small;
public:MKAverage(int m, int k) {this->m = m;this->k = k;}void addElement(int num) {if(q.size() < m){q.push(num);big.insert(num);//插入大的里面process();//处理3个set，保持有序，以及控制数量}else//满了{int tp = q.front();q.pop();q.push(num);big.insert(num);//插入大的里面if(mid.find(tp) != mid.end()){sum -= tp;mid.erase(mid.find(tp));}else if(big.find(tp) != big.end())big.erase(big.find(tp));else // (small.find(tp) != small.end())small.erase(small.find(tp));process();//处理3个set，保持有序，以及控制数量}}void process(){   // 处理 3个 setwhile(big.size() > k){   // big 满了， 小的 放到 small 里small.insert(*big.begin());big.erase(big.begin());}while(small.size() > k){   // small 满了，大的放到 mid 里mid.insert(*small.begin());sum += *small.begin(); // mid 和 增加small.erase(small.begin());}if(big.size()+mid.size()+small.size() < m)return;//数量不够 m 个， 结束逻辑if(*big.begin() < *mid.rbegin()){ // 大的最小的 小于 中间的最大的， 交换数值int v1 = *mid.rbegin(), v2 = *big.begin();sum += v2-v1;mid.erase(--mid.end());big.erase(big.begin());mid.insert(v2);big.insert(v1);}if(big.size() < k){   // 大的数量不够，中间的挪一个最大的到 大的auto it = --mid.end();sum -= *it;big.insert(*it);mid.erase(it);}if(*small.begin() > *mid.begin()){   // 小的最大的 大于 中间最小的，交换数值int v1 = *mid.begin(), v2 = *small.begin();sum += v2-v1;mid.erase(mid.begin());small.erase(small.begin());mid.insert(v2);small.insert(v1);}if(small.size() < k){   // 小的数量不够，中间挪一个最小的到 小的auto it = mid.begin();sum -= *it;small.insert(*it);mid.erase(it);}}int calculateMKAverage() {if(q.size() < m)return -1;return sum/(m-2*k);}
};

560 ms 156.9 MB C++

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！

LeetCode 1825. 求出 MK 平均值（set + queue）相关推荐

定义一个含有30个整型元素的数组，按顺序分别赋予从2开始的偶数；然后按顺序每五个数求出一个平均值，放在另一个数组中并输出。
定义一个含有30个整型元素的数组,按顺序分别赋予从2开始的偶数:然后按顺序每五个数求出一个平均值,放在另一个数组中并输出. var arr = [];for (var i = 1; i <= 6 ...
C语言：向一个文件中写10个数求出其平均值并存入另一个文件中
题目概述: 在文件file0.dat中写10个数求平均值并存入另一个文件file1.dat中编程: #include<stdio.h> #include<stdlib.h> ...
数组赋偶数值并求出平均值
一.题目描述定义一个含有30个元素的整形数组,按顺序分别赋予从2开始的偶数:然后按照顺序每5个数求出一个平均值,放在另一个数组中并输出.(更好的阅读体验,请移步我的个人博客) 二.分析解答本题主要 ...
Python中ArcPy实现对大量长时间序列栅格遥感影像批量逐像元求取像素平均值
本文介绍基于Python中ArcPy模块,对大量长时间序列栅格遥感影像文件的每一个像元进行多时序平均值的求取. 在遥感应用中,我们经常需要对某一景遥感影像中的全部像元的像素值进行平均值求取-- ...
C语言试题二十二之定义了3×3的二维数组，并在主函数中赋值。函数的功能使求出数组周边元素的平均值并作为函数值返回给主函数中的s。
1. 题目定义了3×3的二维数组,并在主函数中赋值.函数的功能使求出数组周边元素的平均值并作为函数值返回给主函数中的s. 2 .温馨提示 C语言试题汇总里可用于计算机二级C语言笔试.机试.研究生复试 ...
C语言（CED）从键盘输入一些整数，求出它们的平均值。数据的个数事先不确定，在输入过程中以ctrl+z键(EOF)作为输入结束的标记。
(请先看置顶博文)https://blog.csdn.net/GenuineMonster/article/details/104495419 要求: 1.从键盘输入一些整数,求出它们的平均值.数据的 ...
编制一个函数jsValue()，求出千位数上的数减百位数上的数减十位数上的数减个位数上的数大于零的个数cnt，再求出所有满足此条件的四位数的平均值pjz1，以及不满足此条件的四位数的平均值pzj2
已知数据文件IN15.DAT中存有300个四位数,并已调用读函数readDat()把这些数存入数组a中,请编制一个函数jsValue(),其功能是:求出千位数上的数减百位数上的数减十位数上的数减个位数 ...
80x86汇编语言循环结构找出最小的偶数并在屏幕上显示求出数组的平均值显示在屏幕上
题目1 写一个完整的80X86汇编语言程序:键盘输入15个数据(转换成数值,存储到一维数组中,数值的长度为字),找出最小的偶数并在屏幕上显示,若没有偶数则显示"没有偶数!". .d ...
LeetCode刷题记录12——232. Implement Queue using Stacks（easy）
LeetCode刷题记录12--232. Implement Queue using Stacks(easy) 目录 LeetCode刷题记录12--232. Implement Queue usin ...

LeetCode 1825. 求出 MK 平均值（set + queue）

文章目录

1. 题目

2. 解题

LeetCode 1825. 求出 MK 平均值（set + queue）相关推荐

最新文章

热门文章