【LeetCode】496 Next Greater Element I

原题地址：496. Next Greater Element I。

解法一：普通解法

//O(n * m)vector<int> nextGreaterElement(vector<int>& findNums, vector<int>& nums) {    vector<int>::iterator pos;    vector<int> res;    int flag;    for (int i = 0; i < findNums.size(); i++) {        pos = find(nums.begin(), nums.end(), findNums[i]);        flag = 0;        for (vector<int>::iterator j = pos + 1; j != nums.end(); j++) {            if (*j > findNums[i]) {                res.push_back(*j);                flag = 1;                break;            }        }        if (flag == 0) {            res.push_back(-1);        }    }    return res;}

解法二：栈解法

使用一个栈来保存递减子序列，当遇到一个数nums[i]大于栈顶元素时，则弹出所有小于nums[i]的元素，并且将他们的Next Greater Element设为nums[i]，随后将nums[i]压栈。

如图所示，nums为[5, 4, 3, 2, 1, 6]，6是前面所有元素[5, 4, 3, 2, 1]的Next Greater Element。

再比如nums为[9, 8, 7, 3, 2, 1, 6]，栈首先会保存一个递减子序列[9, 8, 7, 3, 2, 1]，然后我们看见6大于1，则弹出1 2 3， 他们的Next Greater Element就应该是6。

// O(n)vector<int> nextGreaterElement2(vector<int>& findNums, vector<int>& nums) {    vector<int> res;    stack<int> s;    unordered_map<int, int> map;    for (int i = 0; i < nums.size(); i++) {        while (!s.empty() && s.top() < nums[i]) {            map[s.top()] = nums[i];            s.pop();        }        s.push(nums[i]);    }    for (int i = 0; i < findNums.size(); i++) {        if (map.count(findNums[i]) > 0) {            res.push_back(map[findNums[i]]);        }        else {            res.push_back(-1);        }    }    return res;}

参考：

Java 10 lines linear time complexity O(n) with explanation