Jump Game 的三种思路 - leetcode 55. Jump Game
Jump Game 是一道有意思的题目。题意很简单,给你一个数组,数组的每个元素表示你能前进的最大步数,最开始时你在第一个元素所在的位置,之后你可以前进,问能不能到达最后一个元素位置。
比如:
A = [2, 3, 1, 1, 4], return true.
一种走法是 0 - 2 - 3 - 4
,还有一种走法是 0 - 1 - 4
O(n ^ 2) 解法
一个很显然,几乎不用动脑的解法。
设置一个布尔数组f
,f[0] === true
表示 index === 0
这个位置能够到达,模拟每个位置的前进,最后判断 f[lastIndex]
的值。
/** * @param {number[]} nums * @return {boolean} */var canJump = function(nums) { var f = []; f[0] = true; nums.forEach(function(item, index, array) { if (f[index]) { var tmp = Math.min(array.length - 1, index + item) for (var i = index + 1; i <= tmp; i++) f[i] = true; } }); return f[nums.length - 1] ? false : true;};
但是返回了无情的TLE,给出了一组TLE的数据,数组长度达到了25000,也就是复杂度达到了O(25000 ^ 2),虽然leetcode数据普遍很弱,但是这组TLE也是让我心服口服。
解法1,跪...
O(nlogn) 解法
方法总比困难多,联想到了树状数组中的染色问题。
我们可以把jump的过程看成是染色,还是从左到右枚举位置,比如枚举到 index=0
位置时,nums[0]=5
,也就是说从 index=0
的位置一直可以走到 index=5
的位置,那么我们可以把1~5这一段进行染色。当枚举到 index=1
时,如何判断能不能走到这一步呢?只需求该点被染色的次数,如果大于0,那么就是能到达,然后从该点向后继续染色,最后判断最后一点有没有被染色即可。复杂度 O(nlongn)
。
/** * @param {number[]} nums * @return {boolean} */var sum, n;function lowbit(x) { return x & (-x);}function update(index, val) { while (index) { sum[index] += val; index -= lowbit(index); }}function getAns(index) { var ans = 0; while (index <= n) { ans += sum[index]; index += lowbit(index); } return ans;}var canJump = function(nums) { sum = []; sum[1] = 1; n = nums.reduce(function(pre, item, index) { return Math.max(pre, item + index + 1); }, 0); for (var i = 2; i <= n; i++) sum[i] = 0; for (var i = 0, len = nums.length; i < len; i++) { var isPainted = getAns(i + 1); // 是否被染色 if (!isPainted) continue; update(i + 1 + nums[i], 1); update(i, -1); } return Boolean(getAns(len));};
O(n) 解法
用树状数组显然大材小用了,树状数组可以求得被染色的次数,但是本题只需要判断是否被染色即可;而且本题每次染色都是一次。
进一步思考,我们枚举每一位时都在判断是否被染色过(从而决定是否能够到达该点且能否继续往前走),假设在某一瞬间,index=m
的位置已经被染色了,那么 index=n (n<=m)
的位置肯定已经被染色过了,我们维护一个最右边被染色的点,如果当前枚举点在该点的左侧,那么当前点已经被染色,否则即可停止遍历(因为右边的点再也不可能被染色到了)。
/** * @param {number[]} nums * @return {boolean} */var canJump = function(nums) { var rightMost = 1; for (var i = 0, len = nums.length; i < len; i++) { if (rightMost < i + 1) break; rightMost = Math.max(rightMost, i + 1 + nums[i]); } return rightMost >= len;};