单调栈习题

#include <stack>
#include <unordered_map>
class solution {
public:
	vector<int> func(vector<int>& nums) {
		int len = nums.size();
		vector<int> result(len);
		stack<int> st;
		st.push(0);
		for (int i = 1; i < len; i++) {
			if(nums[i] > nums[st.top()]) {
				while (nums[i] > nums[st.top()] && !st.empty()) {
					result[st.top()] = i - st.top();
					st.pop();
				}				
				st.push(i);
			}
			else {
				st.push(i);
			}
		}
		return result;
	}
};


class solution {
public:
	vector<int> func(vector<int>& nums1,vector<int>&nums2) {
		vector<int> result(nums1.size(), -1);
		stack<int> st;
		st.push(0);
		unordered_map<int, int> umap;
		for (int i = 0; i < nums1.size(); i++) {
			umap.insert(nums1[i], i);
		}
		for (int i = 1; i < nums2.size(); i++) {
			if (nums2[i] > nums2[st.top()]) {
				while (nums2[i] > nums2[st.top()] && !st.empty()) {
					if (umap.find(nums2[st.top()]) != umap.end()) {
						int index = umap[nums2[st.top()]];
						result[index] = nums2[i];
					}
					st.pop();
				}
				st.push(i);
			}
			else {
				st.push(i);
			}			
		}
		return result;
	}
};

//接雨水
class solution {
public:
	int func(vector<int>& nums) {
		int result = 0;
		stack<int> st;
		st.push(0);
		for (int i = 1; i < nums.size(); i++) {
			if (nums[i] > nums[st.top()]) {
				while (!st.empty() && nums[i] > nums[st.top()]) {
					int mid = st.top();
					st.pop();
					if (!st.empty()) {
						int h = min(nums[i], nums[st.top()]) - nums[mid];
						int w = i - st.top() - 1;
						result += h * w;
					}
				}
				st.push(i);
			}
			else {
				st.push(i);
			}
		}
		return result;
	}
};

//柱状图最大面积
//注意在nums的两侧加0
class solution {
public:
	int func(vector<int>& nums) {
		nums.insert(nums.begin(), 0);
		nums.push_back(0);
		stack<int> st;
		int result = 0;
		st.push(0);
		for (int i = 1; i < nums.size(); i++) {
			if (nums[i] < nums[st.top()]) {
				while (nums[i] < nums[st.top()]) {
					int mid = st.top();
					st.pop();
					int left = st.top();
					int w = i - left - 1;
					int h = nums[mid];
					result = max(result, h * w);
				}
				st.push(i);
			}
			else {
				st.push(i);  
			}
		}
	}
};
单调栈，说明栈是单调得，比如单调递增栈，一个序列在push到栈顶之前，必须保证栈里没有再比该序列更小得元素存在了。