Leetcode 457. 环形数组循环 C++

Leetcode 457. 环形数组循环

题目

给定一个含有正整数和负整数的环形数组 nums。 如果某个索引中的数 k 为正数，则向前移动 k 个索引。相反，如果是负数 (-k)，则向后移动 k 个索引。因为数组是环形的，所以可以假设最后一个元素的下一个元素是第一个元素，而第一个元素的前一个元素是最后一个元素。

确定 nums 中是否存在循环（或周期）。循环必须在相同的索引处开始和结束并且循环长度 > 1。此外，一个循环中的所有运动都必须沿着同一方向进行。换句话说，一个循环中不能同时包括向前的运动和向后的运动。

测试样例

示例 1：

输入：[2,-1,1,2,2]
输出：true
解释：存在循环，按索引 0 -> 2 -> 3 -> 0 。循环长度为 3 。

示例 2：

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

示例 3:

输入：[-2,1,-1,-2,-2]
输出：false
解释：按索引 1 -> 2 -> 1 -> ... 的运动无法构成循环，因为按索引 1 -> 2 的运动是向前的运动，而按索引 2 -> 1 的运动是向后的运动。一个循环中的所有运动都必须沿着同一方向进行。

提示：

1. -1000 ≤ nums[i] ≤ 1000
2. nums[i] ≠ 0
3. 0 ≤ nums.length ≤ 5000

题解

暴力
我们用一个数组visit记录是否拜访过
对于nums数组，我们进行修改，进行探索时，我们将其改为INT_MIN，如果再次访问到，表明我们可以以这个下标作为起点；如果不能作为环形数组，我们把它改为0
详细过程见代码

代码

vector<int> visit;
    bool search(vector<int>& nums,int begin,int index,int dir){
        int n = nums.size();
        if(nums[index]==INT_MIN) return true;		//这个下标我们访问过，因此可以以index作为起点
        if(visit[index]==1 || dir*nums[index]<0 || (dir*nums[index])%n==0)   return false;		//前者表明这个结点我们之前尝试过，中间表明方向改变了，后者表明我们原地不动，不符合循环长度>1的条件
        visit[index] = 1;		//进行标记
        int newIndex = index + nums[index];		//查找下一个下标
        if(newIndex >= n) newIndex %= n;
        while(newIndex < 0) newIndex = n+newIndex;
        nums[index] = INT_MIN;		//表明这是我们尝试的下标
        if(search(nums,begin,newIndex,dir)) return true; 
        nums[index] = 0;   			//失败，标记为0
        return false;
    }
    bool circularArrayLoop(vector<int>& nums) {
        int n = nums.size(),dir;
        visit = vector<int>(n,0);
        for(int i=0; i<n; i++){
            if(visit[i])    continue;
            dir = nums[i]>0? 1 : -1;
            if((dir*nums[i])%n != 0){	//不能我们原地不动，否则不符合循环长度>1的条件
                if(search(nums,i,i,dir))  return true;
            }
        }
        return false;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/circular-array-loop
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。