LeetCode Everyday - 最大频率栈

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第16天，点击查看活动详情

最大频率栈

设计一个类似堆栈的数据结构，将元素推入堆栈，并从堆栈中弹出出现频率最高的元素。

实现 FreqStack 类:

• FreqStack() 构造一个空的堆栈。
• void push(int val) 将一个整数 val 压入栈顶。
• int pop() 删除并返回堆栈中出现频率最高的元素。
  • 如果出现频率最高的元素不只一个，则移除并返回最接近栈顶的元素。

示例1:

输入：
["FreqStack","push","push","push","push","push","push","pop","pop","pop","pop"],
[[],[5],[7],[5],[7],[4],[5],[],[],[],[]]
输出：[null,null,null,null,null,null,null,5,7,5,4]
解释：
FreqStack = new FreqStack();
freqStack.push (5);//堆栈为 [5]
freqStack.push (7);//堆栈是 [5,7]
freqStack.push (5);//堆栈是 [5,7,5]
freqStack.push (7);//堆栈是 [5,7,5,7]
freqStack.push (4);//堆栈是 [5,7,5,7,4]
freqStack.push (5);//堆栈是 [5,7,5,7,4,5]
freqStack.pop ();//返回 5 ，因为 5 出现频率最高。堆栈变成 [5,7,5,7,4]。
freqStack.pop ();//返回 7 ，因为 5 和 7 出现频率最高，但7最接近顶部。堆栈变成 [5,7,5,4]。
freqStack.pop ();//返回 5 ，因为 5 出现频率最高。堆栈变成 [5,7,4]。
freqStack.pop ();//返回 4 ，因为 4, 5 和 7 出现频率最高，但 4 是最接近顶部的。堆栈变成 [5,7]。

提示:

• 0 <= val <= 109
• push 和 pop 的操作数不大于 2 * 104。
• 输入保证在调用 pop 之前堆栈中至少有一个元素。

解题思路:

声明两个数组arr和count

• arr的索引为push进来的数据，索引对应的元素表示push/pop操作后，当前索引表示数组的个数
• count为二维数组，与arr作用相反，每一维数据对应当前存在维数的个数的数据集合，即count[[],[1,2,3],[1,2],[1]]表示，1有三个，2有两个，3有一个，每一次push根据存入数据根据其当前个数存入高一维中，即push（2），count变为[[],[1,2,3],[1,2],[1,2]]，push(3),count变为[1,2,3],[1,2，3],[1,2]]，每一次pop操作只需取出count的最后一维的最后一个数据，更新count即可。
• 可以看出，count低维数据对高维数据一定存在包含关系，不必担心连续pop操作后低一维是否为空。每一次push/pop保证了包含关系的单调性和连续性。
• 当某一维数据pop为空后，及时删除该维

我的答案:

var FreqStack = function () {
  this.arr = [];
  this.count = [];
};


FreqStack.prototype.push = function (x) {
  if (this.arr[x]) {
    this.arr[x]++;
  } else {
    this.arr[x] = 1;
  }
  if (this.count[this.arr[x]]) {
    this.count[this.arr[x]].push(x)
  } else {
    this.count[this.arr[x]] = [x]
  }
};

FreqStack.prototype.pop = function () {
  let row = this.count.length;
  let res = 0;
  if (this.count[row - 1].length === 0) {
    this.count.pop()
    res = this.count[row - 2].pop()
  } else {
    res = this.count[row - 1].pop();
  }
  this.arr[res]--;
  return res
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´･ᴗ･` )