leetcode 第713题 乘积小于K的子数组 python解法
题目解析
开始拿到这个题目的时候,我首先想到的是用动态规划来求解,建立一个二位数组dp,其中dp[i][j]代表的是数组中下标为i到j所有的数累乘的结果,状态转移方程也比较好些,两层循环。外层循环遍历整个数组,内层循环遍历外层循环的值到数组最后一位的所有值。dp[i][j] = dp[i][j-1] * nums[j] (j为内层遍历的值)。
这样很快就写出来了,但是提交时(嘣,瞎卡拉卡)— 空间超了限制,因为有二维数组的存在,所以当给的数组很大时,很容易超过空间限制。
后面我看了一下状态转移方程,当前位置的值只与前面一个位置的值有关,而且每个位置的数只会被用过一次,所以完全可以不用建立二维数组,直接用一个临时变量来存储前一个位置的数,这样空间就降到了常数级别。即用post = prev*nums[j],然后判断post是否大于k,如果大于等于k,立刻结束内层循环。后来又想了一下改进的方法,就是在内层遍历的时,先直接从当前位置向后累乘,直到乘积大于等于k时,结束内层循环。最后将结束的位置j减去刚开始的位置i,并将最后的结果加上返回值ret中(即ret+=j-i)。
就这样我又提交了(嘣,瞎卡拉卡)— 时间超了限制,有一个测试用例