給定一個含有正整數和負整數的環形數組 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;}
}