496. Next Greater Element I的C++解法

pascal写多了就有个毛病老是分不清数组的上下界，边界带不带等号半天改不明白。没想出这个题有啥快捷的算法，就老老实实把每个元素都找一遍，时间复杂度O(n²)。注意要判断全空的情况。

class Solution {public:vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {if (findNums.empty()) return findNums;else{int i = 0;for (i; i < findNums.size(); i++){vector<int>::iterator it = find(nums.begin(), nums.end(), findNums[i]);int j = distance(nums.begin(), it)+1;while ((j <= findNums.size()) && (nums[j] < findNums[i])) j = j + 1;if (j >= nums.size()) findNums[i] = -1;else findNums[i] = nums[j];}return findNums;}}};

最优解思路：

遍历nums数组的每个元素，为其找到符合条件的元素。其实就是找一个最长下降序列将其依次压栈，然后从之后的上升序列开始为栈中元素匹配符合的最大值。

class Solution {public:    vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {        stack<int> s;        unordered_map<int, int> m;        for (int n : nums) {            while (s.size() && s.top() < n) {                m[s.top()] = n;                s.pop();            }            s.push(n);        }        vector<int> ans;        for (int n : findNums) ans.push_back(m.count(n) ? m[n] : -1);        return ans;    }};

int n:num即用n遍历num中的每个元素，num中的元素是int类型。