LeetCode 739. 每日温度 题解 C/C++

主要思路

方法一：暴力
维护一个数组 tf[]，始终存储每个温度第一次出现的下标.
从后向前遍历,对每一个温度T[i]，在tf中找T[i]+1到100的第一次出现的最小下标，减去 i 即是要等待的天数，这样可行的原因是我们是从后向前遍历的，tf[]存储的下标始终是在i之后的.

方法二：单调栈
维护一个栈，其中存储的是温度数组下标
从前往后遍历，当前温度高于栈顶温度时，说明栈顶温度的下一个更高的温度就是当前温度，栈顶元素出栈即可，他已经找到了，重复上述操作，直到栈为空或者栈顶温度大于等于当前温度;
如果当前温度没有高于栈顶温度，直接入栈即可.

//方法一：暴力
class Solution {
public:
	vector<int> dailyTemperatures(vector<int>& T) {
		int len = T.size();
		vector<int> ans(len,0),tf(101,INT_MAX);
		for(int i = len-1;i>=0;--i) {
			int idx = INT_MAX;
			for(int j = T[i]+1;j<=100;++j) {
				idx = min(idx,tf[j]);
			}
			if(idx!=INT_MAX) {
				ans[i] = idx-i;
			}
			tf[T[i]] = i;//更新当前温度第一次出现的位置
		}
		return ans;
	}
};

//方法二：单调栈
class Solution {
public:
	vector<int> dailyTemperatures(vector<int>& T) {
		int len = T.size();
		vector<int> ans(len,0);
		stack<int> stk;
		for(int i = 0;i<len;i++) {
			while(!stk.empty()&&T[i]>T[stk.top()]) {
				ans[stk.top()] = i - stk.top();
				stk.pop();
			}
			stk.push(i);
		}
		return ans;
	}
};