给定一个未排序的整数数组 nums , 返回最长递增子序列的个数 。注意这个数列必须是严格递增的。
示例 1:
输入: [1,3,5,4,7]
输出: 2
解释: 有两个最长递增子序列,分别是 [1, 3, 4, 7] 和[1, 3, 5, 7]。
示例 2:
输入: [2,2,2,2,2]
输出: 5
解释: 最长递增子序列的长度是1,并且存在5个子序列的长度为1,因此输出5。
这道题可以和 leetcode 300 最长递增子序列进行对比,300这道是求出序列的长度,而这道题考察这样的序列有多少个。
定义dp[i] 表示以 nums[i] 结尾的最长上升子序列的长度,cnt[i] 表示以 nums[i] 结尾的最长上升子序列的个数。dp[i] 之前,我们已经计算出 dp[0…i−1] 的值,则状态转移方程为:
dp[i]=max(dp[j])+1,其中0≤j<i且num[j]<num[i]
dp[j] 代表 nums[0…j] 中以 nums[j] 结尾的最长上升子序列。
cnt[i] 等于所有满足 dp[j]+1=dp[i] 的 cnt[j] 之和。
时间复杂度:O(n2)
空间复杂度:O(n)
class Solution:
def findNumberOfLIS(self, nums: List[int]) -> int:
n = len(nums)
dp = [1] * n
cn = [1] * n
max_len = 1
max_cnt = 1
for i in range(1, n):
for j in range(i):
if nums[i] > nums[j]:
if dp[j] + 1 > dp[i]:
dp[i] = dp[j] + 1
cn[i] = cn[j]
elif dp[j] + 1 == dp[i]:
cn[i] += cn[j]
if dp[i] > max_len:
max_len = dp[i]
max_cnt = cn[i]
elif dp[i] == max_len:
max_cnt += cn[i]
return max_cnt