循环数组中的环

"""
设计：Python程序设计
作者：初学者
日期：2022年 03月 28日
"""


# 例89  循环数组中的环
#     1.问题描述
#     一个数组包含正整数和负整数。如果某个位置为正整数n，从这个位置出发正向（向右）移动n步；
#     反之，如果某个位置为负整数-n，则从这个位置出发反向（向左）移动n步，数组被视为首尾相连的，
#     即第一个元素视为在最后一个元素的右边，最后一个元素视为在第一个元素的左边。 判断其中是否
#     包含环，即从某一个确定的位置出发，在经过若干次移动后仍能回到这个位置。环必须包含1个以上元素，
#     且必须是单向（不是正向就是反向）移动的。
#     2.问题示例
#     输入[2,-1,1,2,2],输出Ture，表示存在一个环，其下表可以表示为0-->2-->3-->0.
#     3.代码实现
class Solution:
    """
    参数nums：数组
    返回值：返回不布尔类型
    """

    def gte_index(self, i, nums):
        n = (i + nums[i]) % len(nums)
        return n if n >= 0 else n + len(nums)

    def surround(self, nums):
        for i in range(len(nums)):
            if nums[i] == 0:
                continue
            j, k = i, self.gte_index(i, nums)
            while nums[k] * nums[i] > 0 and nums[self.gte_index(k, nums)] * nums[i] > 0:
                if j == k:
                    if j == self.gte_index(j, nums):
                        break
                    return True
                j = self.gte_index(j, nums)
                k = self.gte_index(self.gte_index(k, nums), nums)
            j = i
            while nums[j] * nums[i] > 0:
                next = self.gte_index(j, nums)
                nums[j] = 0
                j = next
        return False


# 主函数
if __name__ == '__main__':
    s = Solution()
    s1 = Solution()
    m = [4, 5, 8, -4]
    n = [2, -1, 1, 2, 2]
    print("输入数组：", n)
    print("输出：", s.surround(n))
    print("输出：", s1.surround(m))