给定一个含有正整数和负整数的环形数组 nums。 如果某个索引中的数 k 为正数,则向前移动 k 个索引。相反,如果是负数 (-k),则向后移动 k 个索引。因为数组是环形的,所以可以假设最后一个元素的下一个元素是第一个元素,而第一个元素的前一个元素是最后一个元素。
确定 nums 中是否存在循环(或周期)。循环必须在相同的索引处开始和结束并且循环长度 > 1。此外,一个循环中的所有运动都必须沿着同一方向进行。换句话说,一个循环中不能同时包括向前的运动和向后的运动。
示例 1:
输入:[2,-1,1,2,2]
输出:true
解释:存在循环,按索引 0 -> 2 -> 3 -> 0 。循环长度为 3 。
代码
class Solution {public boolean circularArrayLoop(int[] nums) {int n=nums.length;boolean[] flags=new boolean[n];Set<Integer> set=new HashSet<>();for(int i=0;i<n;i++)//检查正数{if(nums[i]<0||flags[i]) continue;set.add(i);int pre=i;while (true)//向后移动{int next=(pre+nums[pre])%n;if(nums[next]<0||next==pre||flags[next]){
//因为路径是唯一的,所以路径上一旦有不满足的点,就整条路径标记起来,再也不用遍历这些点for(int c:set)flags[c]=true;set.clear();break;}if(set.contains(next)) return true;//可以产生环set.add(next);pre=next;//跳到下一点}}for(int i=0;i<n;i++)//检查负数{if(nums[i]>0||flags[i]) continue;set.add(i);int pre=i;while (true){int next=(pre+nums[pre])%n;next=next<0?next+n:next;if(nums[next]>0||next==pre||flags[next]){for(int c:set)flags[c]=true;set.clear();break;}if(set.contains(next)) return true;set.add(next);pre=next;}}return false;}
}