天天看點

LeetCode503. 下一個更大元素 II

給定一個循環數組(最後一個元素的下一個元素是數組的第一個元素),輸出每個元素的下一個更大元素。數字 x 的下一個更大的元素是按數組周遊順序,這個數字之後的第一個比它更大的數,這意味着你應該循環地搜尋它的下一個更大的數。如果不存在,則輸出 -1。

示例 1:

輸入: [1,2,1]

輸出: [2,-1,2]

解釋: 第一個 1 的下一個更大的數是 2;

數字 2 找不到下一個更大的數;

第二個 1 的下一個最大的數需要循環搜尋,結果也是 2。

注意: 輸入數組的長度不會超過 10000。

來源:力扣(LeetCode)

連結:https://leetcode-cn.com/problems/next-greater-element-ii

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        int n=nums.size();
        vector<int> ans;
        for(int i=0;i<n;i++){
            int tmp=-1;
            for(int j=(i+1)%n,k=0;k<n;(j++)%n,k++){
                if(nums[j%n]>nums[i]){
                    tmp=nums[j%n];
                    break;
                }
            }
            ans.push_back(tmp);
        }
        return ans;
    }
};