DP类型的题目,我们用DP[0][i]表示在i处结尾的LIS的长度,DP[1][i]表示在i处结尾的LIS的数量。DP方程为:
- DP[0][i] = max(DP[0][j]) + 1 (0 <= j < i && nums[j] < nums[i])
- DP[1][i] = sum(DP[1][j]) where DP[0][j] = DP[0][i] - 1
由此我们可以写出代码,时间复杂度O(n^2),空间复杂度O(n):
另一种方法,这道题可以用segment tree来做。segment tree存的区间是值域上的区间而不是以index来划分。我们要存的东西比较有意思,对于当前节点和其所表示的区间[s, e],从所有以处于[s, e]区间范围中的元素结尾的子序列当中,找出最长的子序列长度和数量,存入node中。
插入的话,我们左向右扫数组,对于每一个数nums[i],我们查询以[minVal, nums[i - 1]]范围中的数结尾的最长的子序列的长度和数量。那么我们可以根据查询的结果更新segment tree,比如我们得知以小于nums[i]结尾的子序列的长度为len, 数量为cnt。那么我们要在segment tree中把以nums[i]结尾的最长子序列的长度和数量更新为len + 1和cnt。
我们每次更新的时候,只需要去左右两个子区间:
- 如果左子区间最大长度比右子区间长,我们取左子区间的长度和数量
- 如果左子区间最大长度比右子区间短,我们取右子区间的长度和数量
- 如果左子区间最大长度和右子区间相等,我们取任意区间的长度,数量为两个子区间之和
实现的时候,对于长度为N的数组,我们要把值域map到[0, N - 1]的区间,这样可以省下很多空间。时间复杂度O(N *log N),代码如下:
区间查询的话Binary Indexed Tree当然也可以做,虽然和我们在链接文章中分析的,BIT实现Range Min/Max Query的话会比普通的BIT复杂一些,但是这一道题,对于每一个nums[i]我们只插入不更新,所以简单的BIT即可实现。时间复杂度O(N *log N),代码如下:
No comments:
Post a Comment