题目来源:https://leetcode-cn.com/problems/circular-array-loop/
大致题意:
给定一个数组,数组中每个元素值都不为0,且元素值可以理解为当前元素有一个指向 当前元素下标+元素值 位置(若小于0,则位置可以理解为 加上数组长度n 后的位置,若大于等于n,位置即为 对n取余)的指针。
判断数组中元素的指向是否构成环(环内指向方向一致,且构成循环)。
思路
数组元素的值可正可负,也可能大于数组长度。
- 当元素值+下标 小于0时,对n取余(先对n取余是防止负数过小)加上n即为指向位置
- 当元素值+下标 大于等于n时,对n取余即为指向位置。
所以求指向方法为:
- 当前位置为pos,数组为nums,长度为n
- ((pos + nums[pos])% n + n) % n
可以使用快慢指针的方法找环。
快慢指针:使用两个指针对路径进行遍历,初始时慢指针在起点,快指针在起点之后一个位置。慢指针每次移动一步,快指针每次移动两步。如果路径构成环,那么快指针定然先进入环,且在环中进行循环,并且再慢指针进入环后,快指针与慢指针一定会在有限的步数内相遇。
在使用标记数组标记查找过的路径,避免重复查找。
标记可以直接在原数组进行,直接将经过的位置记为0即可。
代码:
public boolean circularArrayLoop(int[] nums) {
int n = nums.length;
for (int i = 0; i < n; i++) {
// 如果当前位置元素为0,代表当前节点已经查找过
if (nums[i] == 0) {
continue;
}
int slow = i;
int fast = next(nums, i);
// 当slow对应位置与后两个位置指向相同,且都不为0时
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)); // 快指针移动两步
}
int path = i;
// 将本轮查找路径点标记
while (nums[path] * nums[next(nums, path)] > 0) {
nums[path] = 0;
path = next(nums, path);
}
}
return false;
}
public int next(int[] nums, int pos) {
int n = nums.length;
// 保证范围在[0,n)
return ((pos + nums[pos]) % n + n) % n;
}