编程题-算法-中等-leetcode-739. 每日温度

编程题-算法-中等-leetcode-739. 每日温度

  • 1、题目:每日温度
    • 2.1 方式一
    • 2.1 方式二:

1、题目:每日温度

编程题-算法-中等-leetcode-739. 每日温度_第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;
    }
}

提交后的结果如下图,可以看到时间复杂度和空间复杂度都比较好。
编程题-算法-中等-leetcode-739. 每日温度_第2张图片

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;
    }
}

编程题-算法-中等-leetcode-739. 每日温度_第3张图片

你可能感兴趣的:(算法基础,leetcode,算法)