请根据每日 气温
列表 temperatures
,重新生成一个列表,要求其对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0
来代替。
解法一:暴力算法(超时)
int* dailyTemperatures(int* temperatures, int temperaturesSize, int* returnSize){
int* day;
*returnSize=temperaturesSize;
day=(int*)malloc(temperaturesSize*sizeof(int));
for(int i=0;i<temperaturesSize;i++)
for(int j=i+1;j<temperaturesSize;j++)
{
if(temperatures[i]<temperatures[j])
{
day[i]=j-i;
break;
}
else day[i]=0;
}
day[temperaturesSize-1]=0;
return day;
}
解法2:单调栈
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* dailyTemperatures(int* T, int TSize, int* returnSize)
{
int * num = (int *)malloc(sizeof(int) * TSize);
int * res = (int *)malloc(sizeof(int) * TSize);
memset(res, 0, sizeof(int) * TSize);
int top = -1;
for (int i = 0; i < TSize; i++)
{
while (top >= 0 && T[i] > T[num[top]])
{
res[num[top]] = i - num[top];
top--;
}
num[++top] = i;
}
*returnSize = TSize;
return res;
}