题目描述
给出连续若干天的气温数据如[73, 74, 75, 71, 69, 72, 76, 73]
,输出多少天后气温会高于当天,上面例子的输出结果为[1, 1, 4, 2, 1, 1, 0, 0]
。温度的范围从30到100,数据长度1到30000。
蒟蒻的解法
public int[] dailyTemperatures(int[] T) {
int[] result = new int[T.length]; // 存储最后结果
// tempData在搜索比自身大的数可以按顺序搜索
int[] tempData = new int[71]; // 长度71是因为数据范围是30-100,里面存储每个数最近一次出现时的下标
int[] resultCopy = new int[71]; // 存储对应数据的结果值
// 初始化
for (int i = 0; i < tempData.length; i++) {
tempData[i] = -1;
resultCopy[i] = -1;
}
for (int i = 0; i < T.length; i++) {
// 如果该数值还未在tempData中出现过
if (tempData[T[i] - 30] < i) {
tempData[T[i] - 30] = i;
}
// 处理大多数数字相同的情况[76,76.......],没添加的时候超时了(真滴菜)
if (resultCopy[T[i] - 30] == 0) {
result[i] = 0;
continue;
}
// 找到相同数值的数的结果,如果大于当前值的下标时,将其作为搜索时的终止条件
if (resultCopy[T[i]- 30] > i) {
result[i] = resultCopy[T[i] - 30] - i;
continue;
}
// 如果该数值之前没有出现过,从这个数的位置到tempData末端寻找之前出现过大于它且下标也大于它的最近的数
// -1表示不存在
int getBigNumIndex = getBiggerNum(tempData, T[i], i, tempData.length - 1);
if (getBigNumIndex != -1) {
result[i] = getBigNumIndex - i;
resultCopy[T[i] - 30] = getBigNumIndex;
} else {
// 不存在的情况下去原数组往后搜索第一个大于该数的值的下标作为结果
boolean findBig = false;
for (int j = i + 1; j < T.length; j++) {
tempData[T[j] - 30] = j;
if (T[j] > T[i]) {
result[i] = j - i;
resultCopy[T[i] - 30] = j;
findBig = true;
break;
}
}
// 没找到比它大的置0
if (!findBig) {
result[i] = 0;
resultCopy[T[i] - 30] = 0;
}
}
}
return result;
}
/**
* 在source中的start和end范围内找最近的比target大的数
**/
public int getBiggerNum(int[] source, int target, int start, int end) {
int minMax = -1;
for (int i = target - 30; i <= end; i++) {
if (source[i] > start) {
if (minMax > source[i]) {
minMax = source[i];
}
}
}
return minMax;
}
菊苣的解法
public int[] dailyTemperaturesUsingStack(int[] temperatures) {
Stack<Integer> stack = new Stack<>();
int[] ret = new int[temperatures.length];
for(int i = 0; i < temperatures.length; i++) {
while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]) {
int idx = stack.pop();
ret[idx] = i - idx;
}
stack.push(i);
}
return ret;
}
每次看到别人的解法就觉得自己用那么长时间写了那么一大堆像个智障。
按样例输入[73, 74, 75, 71, 69, 72, 76, 73]
初始时:stack=[]
i = 0:stack为空,stack插入0,此时stack=[0]
i = 1:temperatures[1] > temperatures[0],ret[0] = 1 - 0,stack=[1]
i = 2:temperatures[2] > temperatures[1],ret[1] = 2 - 1,stack=[2]
i = 3:temperatures[3] < temperatures[2],stack=[2,3]
i = 4:temperatures[4] < temperatures[3],stack=[2,3,4]
i = 5:temperatures[5] > temperatures[4],ret[4] = 1, stack=[2, 3]
,temperatures[5] > temperatures[3],ret[3] = 2, stack=[2]
, stack.push(5),stack=[2,5]
i = 6:temperatures[6] > temperatures[5],ret[5] = 1,stack=[2]
,temperatures[6] > temperatures[2],ret[2] = 4,stack=[]
,stack.push(6),stack=[6]
i = 7:temperatures[7] < temperatures[6]
循环结束:这里用到了java数组的一个特性,创建数组没有手动复制的话默认值为0