leetcode 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

分析：先用一个 map 记录出现过的所有数字，接着遍历 nums 数组，循环检查以 numsi 作为子集中最中间的数字时，是否符合条件，若符合，则子集长度 +2；不符合则跳出循环。检查完之后，需要特判一下出现的 1 的次数，取遍历时获得的子集最大值，与小于等于 1 出现次数的最大奇数值作为答案。

class Solution {
public:
    int maximumLength(vector<int>& nums) {
        int n=nums.size(),ans=1;
        map<int,int>mp;
        for(int i=0;i<n;++i)
            mp[nums[i]]++;
        for(int i=0;i<n;++i)
        {
            int val=sqrt(nums[i]),cnt=1,temp=nums[i];
            while(val*val==temp&&val!=1)
            {
                if(mp[val]>=2)cnt+=2;
                else break;
                temp=val,val=sqrt(val);
            }
            ans=max(ans,cnt);
        }
        if(mp[1])ans=max(ans,mp[1]%2==1?mp[1]:mp[1]-1);
        
        return ans;
    }
};