leetcode457. 环形数组循环

给定一个含有正整数和负整数的环形数组 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;

    }
}