1、题目:每日温度
2、编程解答
看到题目,可以想到用一个双向队列来记录对应的温度和下标值。然后遍历温度temperatures,当遍历的温度值(temp)比队首元素的温度(firstTempInQue)高,则可以得到队首元素的结果。
如果当前温度值(temp)比队首元素的温度(firstTempInQue)低,则将其放到队首。
2.1 方式一
方式一比较好理解,自定义类Node和队列,进行解答
class Node {
int temperature;
int index;
public Node(int tmp, int idx) {
this.temperature = tmp;
this.index = idx;
}
}
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] res = new int[temperatures.length];
Deque<Node> queue = new LinkedList<>();
for (int i=0; i<temperatures.length; i++) {
Node node = new Node(temperatures[i], i);
// 如果队列非空,且对首元素温度低于当前遍历的温度值,则找到队首对应的结果
while (!queue.isEmpty() && queue.peekFirst().temperature<node.temperature) {
Node firstNode = queue.pollFirst();
res[firstNode.index] = node.index - firstNode.index;
}
// 队列为空,或者队首的温度大于当前遍历温度值,则将当前温度值放在队首
queue.offerFirst(node);
}
// 队列中剩下的元素,代表找不到温度比他们高的值,因此结果都为0
return res;
}
}
提交后的结果如下图,可以看到时间复杂度和空间复杂度都比较好。
2.1 方式二:
方式二是不借助自定义的类Node,而是将队列里保存对应温度的下标值,通过下标来进行判断,当然原理其实和方式一基本一致。理解了方式一的话,方式二也就很好理解了。
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int length = temperatures.length;
int[] ans = new int[length];
Deque<Integer> stack = new LinkedList<Integer>();
for (int i = 0; i < length; i++) {
int temperature = temperatures[i];
while (!stack.isEmpty() && temperature > temperatures[stack.peek()]) {
int prevIndex = stack.pop();
ans[prevIndex] = i - prevIndex;
}
stack.push(i);
}
return ans;
}
}