3020. 子集中元素的最大数量

简介: 3020. 子集中元素的最大数量

说在前面

🎈不知道大家对于算法的学习是一个怎样的心态呢?为了面试还是因为兴趣?不管是出于什么原因,算法学习需要持续保持。

题目描述

给你一个 正整数 数组 nums

你需要从数组中选出一个满足下述条件的子集

  • 你可以将选中的元素放置在一个下标从 0 开始的数组中,并使其遵循以下模式:[x, x2, x4, ..., xk/2, xk, xk/2, ..., x4, x2, x]注意k 可以是任何 非负 的 2 的幂)。例如,[2, 4, 16, 4, 2][3, 9, 3] 都符合这一模式,而 [2, 4, 8, 4, 2] 则不符合。

返回满足这些条件的子集中,元素数量的 最大值

示例 1:

输入: nums = [5,4,1,2,2]
输出: 3
解释: 选择子集 {4,2,2} ,将其放在数组 [2,4,2] 中,它遵循该模式,且 22 == 4 。因此答案是 3 。

示例 2:

输入: nums = [1,3,2,4]
输出: 1
解释: 选择子集 {1},将其放在数组 [1] 中,它遵循该模式。因此答案是 1 。注意我们也可以选择子集 {2} 、{4} 或 {3} ,可能存在多个子集都能得到相同的答案。

提示:

  • 2 <= nums.length <= 10^5
  • 1 <= nums[i] <= 10^9

解题思路

先对nums数组按从小到大进行排序,使用一个哈希表来统计每个数字出现的次数。

计算最大子集元素数量:

const getMax = (num) => {
  if (num === 1) {
    if (map[num] % 2 === 0) return map[num] - 1;
    return map[num];
  }
  let cnt = 0;
  while (map[num] > 1) {
    flag[num] = true;
    cnt += 2;
    num = Math.pow(num, 2);
  }
  if (map[num] === 1) return cnt + 1;
  return cnt - 1;
};

这里我们需要对元素1进行特殊处理,因为1的任何次幂都是1,所以奇数个1也可以组成一个满足条件的子集。其他元素则需要对其进行求幂计算,找到存在的连续的最高次幂元素。并对计算过的元素做一个标记,避免进行重复的计算。

AC代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var maximumLength = function (nums) {
  nums = nums.sort((a, b) => a - b);
  const map = {};
  const flag = {};
  for (const num of nums) {
    const val = map[num] || 0;
    map[num] = val + 1;
  }
  let res = 1;
  const getMax = (num) => {
    if (num === 1) {
      if (map[num] % 2 === 0) return map[num] - 1;
      return map[num];
    }
    let cnt = 0;
    while (map[num] > 1) {
      flag[num] = true;
      cnt += 2;
      num = Math.pow(num, 2);
    }
    if (map[num] === 1) return cnt + 1;
    return cnt - 1;
  };
  for (const k of nums) {
    if (!flag[k]) res = Math.max(getMax(k), res);
  }
  return res;
};

公众号

关注公众号『前端也能这么有趣』,获取更多有趣内容。

说在后面

🎉 这里是 JYeontu,现在是一名前端工程师,有空会刷刷算法题,平时喜欢打羽毛球 🏸 ,平时也喜欢写些东西,既为自己记录 📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解 🙇,写错的地方望指出,定会认真改进 😊,偶尔也会在自己的公众号『前端也能这么有趣』发一些比较有趣的文章,有兴趣的也可以关注下。在此谢谢大家的支持,我们下文再见 🙌。

目录
相关文章
|
6天前
13.从M个球中取出N个球的所有组合情况
13.从M个球中取出N个球的所有组合情况
9 0
|
8天前
|
索引
将数组指定索引位置的元素 移动到 目标索引位置,且不改变其他元素原本的顺序,注意这个不是对调元素位置,是移动某一个元素位置不影响其他元素顺(使用场景:拖拽改变数据的顺序,点击上下左右箭头移动元素顺序)
将数组指定索引位置的元素 移动到 目标索引位置,且不改变其他元素原本的顺序,注意这个不是对调元素位置,是移动某一个元素位置不影响其他元素顺(使用场景:拖拽改变数据的顺序,点击上下左右箭头移动元素顺序)
|
8天前
|
算法 Java C++
请实现一个队列,支持以下操作:添加元素、删除第一个元素、获取第一个元素。
请实现一个队列,支持以下操作:添加元素、删除第一个元素、获取第一个元素。
19 0
|
8天前
|
C++
『C/C++』Eg4: 求自定类型元素的平均
『C/C++』Eg4: 求自定类型元素的平均
|
6月前
|
JavaScript Python
从列表中或数组中随机抽取固定数量的元素组成新的数组或列表
从列表中或数组中随机抽取固定数量的元素组成新的数组或列表
40 0
|
11月前
|
存储
返回集合中最大,最小的元素,再将元素进行排序
返回集合中最大,最小的元素,再将元素进行排序
49 0
|
11月前
|
安全 Java
子类成员数目一定大于等于父类成员数目
子类成员数目一定大于等于父类成员数目
96 0
python实现将给定列表划分为元素和大致相等的两个子列表
python实现将给定列表划分为元素和大致相等的两个子列表
|
测试技术
List分割:等分成n个list;指定元素个数n等分
List分割:等分成n个list;指定元素个数n等分
539 0