Leetcode 2862. Maximum Element-Sum of a Complete Subset of Indices

  • [Leetcode 2862. Maximum Element-Sum of a Complete Subset of Indices](#Leetcode 2862. Maximum Element-Sum of a Complete Subset of Indices)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题的核心在于想明白一点:

  • 要使得子序列当中任意两个数之积均为平方数,那么子序列当中的所有数必然都是一系列平方数的某一个公倍数。

因此,我们只需要不超过数组长度 n n n的所有平方数,然后分别将其扩展倍数即可。

而对于扩展倍数之后依然有效的平方数,我们同样可以通过二分法进行优化寻找。

2. 代码实现

给出python代码实现如下:

python 复制代码
class Solution:    
    def maximumSum(self, nums: List[int]) -> int:
        n = len(nums)

        completes = [i*i for i in range(1, int(sqrt(n) + 2)) if i * i <= n]
        res = max(max(nums), sum([nums[i-1] for i in completes]))
        for p in range(1, n+1):
            if p > n:
                break
            if completes[-1] * p > n:
                i, j = 0, len(completes)-1
                while j-i>1:
                    m = (i+j)//2
                    if completes[m] * p > n:
                        j = m
                    else:
                        i = m
            else:
                j = len(completes)
            if j == 1:
                break
            s = sum([nums[p*i-1] for i in completes[:j]])
            res = max(res, s)
        return res

提交代码评测得到:耗时290ms,占用内存18.1MB。

相关推荐
Pluchon5 分钟前
硅基计划4.0 算法 二叉树深搜(DFS)
java·数据结构·算法·leetcode·深度优先·剪枝
sprintzer1 小时前
10.6-10.15力扣模拟刷题
算法·leetcode·职场和发展
一匹电信狗2 小时前
【C++】C++风格的类型转换
服务器·开发语言·c++·leetcode·小程序·stl·visual studio
学学学无无止境3 小时前
力扣-上升的温度
leetcode
Emilia486.5 小时前
【Leetcode&nowcode&数据结构】顺序表的应用
数据结构·算法·leetcode
Dream it possible!6 小时前
LeetCode 面试经典 150_栈_简化路径(53_71_C++_中等)(栈+stringstream)
c++·leetcode·面试·
程序员阿鹏6 小时前
49.字母异位词分组
java·开发语言·leetcode
Espresso Macchiato7 小时前
Leetcode 3715. Sum of Perfect Square Ancestors
算法·leetcode·职场和发展·leetcode hard·树的遍历·leetcode 3715·leetcode周赛471
Miraitowa_cheems13 小时前
LeetCode算法日记 - Day 73: 最小路径和、地下城游戏
数据结构·算法·leetcode·职场和发展·深度优先·动态规划·推荐算法
野蛮人6号13 小时前
力扣热题100道之560和位K的子数组
数据结构·算法·leetcode