给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指在第 i 天之后,才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。
示例 1:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
示例 2:
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]
示例 3:
输入: temperatures = [30,60,90]
输出: [1,1,0]
提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一开始想用暴力法,又超时啦!!
这个题需要用于一个栈将循环向前的过程中小于前面一个数据的则进栈,如果又大于栈顶的数据就出栈。
出栈的时候要循环,不一定只大于栈顶对应的值,也可能大于多个。
最后循环晚所有数据之后,要把没有出栈的全部置0.
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* dailyTemperatures(int* temperatures, int temperaturesSize, int* returnSize){
*returnSize = temperaturesSize;
int *retArr = (int *)malloc(sizeof(int) * temperaturesSize);
int stack[100000] = {0};
int top = -1;
for (int i = 0; i < temperaturesSize; i++) {
if (top < 0) {
stack[++top] = i;
} else if (temperatures[i] <= temperatures[stack[top]]){
stack[++top] = i;
} else {
while (top >= 0 && temperatures[i] > temperatures[stack[top]]) {
retArr[stack[top]] = i - stack[top];
stack[top] = -1;
top--;
}
stack[++top] = i;
}
}
while (top >= 0) {
retArr[stack[top--]] = 0;
}
return retArr;
}
/*
暴力法可以解出来,但是超时
int* dailyTemperatures(int* temperatures, int temperaturesSize, int* returnSize){
*returnSize = temperaturesSize;
int *retArr = (int *)malloc(sizeof(int) * temperaturesSize);
for (int i = 0; i < temperaturesSize; i++) {
retArr[i] = 0;
for (int j = i + 1; j < temperaturesSize; j++) {
if (temperatures[j] > temperatures[i]) {
retArr[i] = j - i;
break;
}
}
}
return retArr;
}
*/