题解 | #奶牛的下一个元素#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param nums int整型一维数组 
# @return int整型一维数组
#
class Solution:
    def nextGreaterCowWeight(self , nums: List[int]) -> List[int]:
        # write code here
        n = len(nums)
        result = [-1] * n  # 初始化结果列表为-1
        stack = []  # 单调栈，存储元素下标

        # 从右向左遍历循环数组
        for i in range(2 * n - 1):
            while stack and nums[i % n] > nums[stack[-1]]:
                j = stack.pop()
                result[j % n] = nums[i % n]  # 当前元素的下一个更大的体重
            stack.append(i % n)  # 当前元素下标入栈

        return result