swift 算法题-每日温度

1. 题目说明

根据每日 气温 列表，请重新生成一个列表，对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高，请在该位置用 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度，都是在 [30, 100] 范围内的整数。

试题来源：力扣

2. 解题思路

使用一个数值降序的栈, 利用下标的差值来计算

3. 实现代码

struct Stack<T>{
    private var array:[T] = []
    
    mutating func push(_ ele:T) {
        array.append(ele)
    }
    
    mutating func pop() -> T? {
        return array.popLast()
    }
    
    func top() -> T? {
        return array.last
    }
    
    func isEmpty() -> Bool {
        return array.count == 0
    }
}

class Solution {
    func dailyTemperatures(_ T: [Int]) -> [Int] {
        let len = T.count
        var res = [Int](repeating: 0, count: len)
        var s:Stack<Int> = Stack()
        for i in 0 ..< len {
            
            while(!s.isEmpty() && T[i] > T[s.top()!]){
                let val = s.pop()!
                res[val] = i - val
            }
            s.push(i)
        }
        return res
    }
}

4. 测试代码

var nums1 = [73, 74, 75, 71, 69, 72, 76, 73]

let s = Solution()

let res = s.dailyTemperatures(nums1)

print(res)

运行结果:

[1, 1, 4, 2, 1, 1, 0, 0]