【Leetcode每日一题】1646. 获取生成数组中的最大值

题目

给你一个整数 n 。按下述规则生成一个长度为 n + 1 的数组 nums ：

• nums[0] = 0
• nums[1] = 1
• 当 2 <= 2 * i <= n 时，nums[2 * i] = nums[i]
• 当 2 <= 2 * i + 1 <= n 时，nums[2 * i + 1] = nums[i] + nums[i + 1]

返回生成数组 nums 中的 最大 值。

示例

示例1

输入： n = 7
输出： 3
解释： 根据规则：
nums[0] = 0
nums[1] = 1
nums[(1 * 2) = 2] = nums[1] = 1
nums[(1 * 2) + 1 = 3] = nums[1] + nums[2] = 1 + 1 = 2
nums[(2 * 2) = 4] = nums[2] = 1
nums[(2 * 2) + 1 = 5] = nums[2] + nums[3] = 1 + 2 = 3
nums[(3 * 2) = 6] = nums[3] = 2
nums[(3 * 2) + 1 = 7] = nums[3] + nums[4] = 2 + 1 = 3
因此，nums = [0,1,1,2,1,3,2,3]，最大值 3

示例2

输入： n = 2
输出： 1
解释： 根据规则，nums[0]、nums[1] 和 nums[2] 之中的最大值是 1

关键思路

根据题目所给公式写代码即可。

代码实现

class Solution(object):
    def getMaximumGenerated(self, n):
        """
        :type n: int
        :rtype: int
        """

        nums = [0,1]

        for i in range(2,n+1):
            if i%2 == 0: 
                nums.append(nums[i/2])
            else:
                nums.append(nums[(i-1)/2] + nums[(i-1)/2 + 1])

        return max(nums)

if __name__ == "__main__":
    n = input()
    obj = Solution()
    result = obj.getMaximumGenerated(n)
    print(result)

运行结果

3
1

链接

https://leetcode-cn.com/problems/get-maximum-in-generated-array