LeetCode 1646. 获取生成数组中的最大值 Python

题目来源：https://leetcode-cn.com/problems/get-maximum-in-generated-array/

题目描述

给你一个整数 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 中的 最大 值。

示例

输入：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

解题思路

按照奇偶数进行求解

代码

n = int(input())
nums=[0]*(n+1)
if n < 2:
    print(n)
else:
    nums[1] = 1
    for i in range(2,n+1):
        if i % 2 == 0:
            nums[i] = nums[i//2]
        else:
            nums[i] = nums[i//2] + nums[i//2+1]
    print(max(nums))

如有错漏之处，敬请指正！