日拱算法:环形数组是否存在循环

「这是我参与2022首次更文挑战的第15天,活动详情查看:2022首次更文挑战


题:

存在一个不含 0 的 环形 数组 nums ,每个 nums[i] 都表示位于下标 i 的角色应该向前或向后移动的下标个数:

  • 如果 nums[i] 是正数,向前(下标递增方向)移动 |nums[i]| 步

  • 如果 nums[i] 是负数,向后(下标递减方向)移动 |nums[i]| 步

因为数组是 环形 的,所以可以假设从最后一个元素向前移动一步会到达第一个元素,而第一个元素向后移动一步会到达最后一个元素。

数组中的 循环 由长度为 k 的下标序列 seq 标识:

  • 遵循上述移动规则将导致一组重复下标序列 seq[0] -> seq[1] -> ... -> seq[k - 1] -> seq[0] -> ...

  • 所有 nums[seq[j]] 应当不是 全正 就是 全负

  • k > 1

如果 nums 中存在循环,返回 true ;否则,返回 false 。

示例 1:

输入:nums = [2,-1,1,2,2]
输出:true
解释:存在循环,按下标 0 -> 2 -> 3 -> 0 。循环长度为 3 。

示例 2:

输入:nums = [-1,2]
输出:false
解释:按下标 1 -> 1 -> 1 ... 的运动无法构成循环,因为循环的长度为 1 。根据定义,循环的长度必须大于 1 。

示例 3:

输入:nums = [-2,1,-1,-2,-2]
输出:false
解释:按下标 1 -> 2 -> 1 -> ... 的运动无法构成循环,因为 nums[1] 是正数,而 nums[2] 是负数。
所有 nums[seq[j]] 应当不是全正就是全负。
复制代码

注:

  • 1 <= nums.length <= 5000
  • -1000 <= nums[i] <= 1000
  • nums[i] != 0

解题思路:

今天的题目意思比较难懂,不懂题目的朋友可以看一下本节的题意分析。

首先,要理解题目中的「环形数组」是什么。「环形数组」就是在逻辑上首尾相接的数组,即最后一个元素和第一个元素在逻辑上是相邻的(在物理存储上仍然是个普通的数组)。

那么环形数组中存在循环是什么意思呢?这就是说,在环形数组中,每个位置存储的元素表示当前位置应该向前/向后移动的步数。如果在环形数组中绕了一圈又回到了原地,那么说明数组中存在循环。

举个例子,在环形数组 [2, -1, 1, 2, 2] 中,存在循环,同时,题目约定了循环的条件:所有 nums[seq[j]] 应当不是 全正 就是 全负,即只能沿着一个方向走;k > 1,即要求环的大小 > 1。

做过「判断链表中是否有环」题目的知道,这两题的思路是一致的,常见的思路就是快慢指针,在链表问题中,快指针每次走 2 步,慢指针每次走 1 步。当快慢指针相遇的时候,说明存在环。

在每次循环的过程中,必须保证所经历过的所有数字都是同号的。

那么,在快指针经历过的每个位置都要判断一下和出发点的数字是不是相同的符号。

当快慢指针相遇的时候,还要判断环的大小不是 1。

所以,找到相遇点的位置后,如果再走 1 步,判断是不是自己。

JavaScript 实现:

var circularArrayLoop = function(nums) {
    const n = nums.length;
    for (let i = 0; i < n; i++) {
        if (nums[i] === 0) {
            continue;
        }
        let slow = i, fast = next(nums, i);
        // 判断非零且方向相同
        while (nums[slow] * nums[fast] > 0 && nums[slow] * nums[next(nums, fast)] > 0) {
            if (slow === fast) {
                if (slow !== next(nums, slow)) {
                    return true;
                } else {
                    break;
                }
            }
            slow = next(nums, slow);
            fast = next(nums, next(nums, fast));
        }
        let add = i;
        while (nums[add] * nums[next(nums, add)] > 0) {
            const tmp = add;
            add = next(nums, add);
            nums[tmp] = 0;
        }
    }
    return false;
}

const next = (nums, cur) => {
    const n = nums.length;
    return ((cur + nums[cur]) % n + n) % n; // 保证返回值在 [0,n) 中
}
复制代码

我是掘金安东尼,输出暴露输入,技术洞见生活,再会~

猜你喜欢

转载自juejin.im/post/7059749140195966989