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;
    }
};