LeetCode-in-Rust
739. Daily Temperatures
Medium
Given an array of integers temperatures represents the daily temperatures, return an array answer such that answer[i] is the number of days you have to wait after the ith day to get a warmer temperature. If there is no future day for which this is possible, keep answer[i] == 0 instead.
Example 1:
Input: temperatures = [73,74,75,71,69,72,76,73]
Output: [1,1,4,2,1,1,0,0]
Example 2:
Input: temperatures = [30,40,50,60]
Output: [1,1,1,0]
Example 3:
Input: temperatures = [30,60,90]
Output: [1,1,0]
Constraints:
1 <= temperatures.length <= 10530 <= temperatures[i] <= 100
Solution
impl Solution {
pub fn daily_temperatures(temperatures: Vec<i32>) -> Vec<i32> {
let mut sol = vec![0; temperatures.len()];
for i in (0..temperatures.len() - 1).rev() {
let mut j = i + 1;
while j < temperatures.len() {
if temperatures[i] < temperatures[j] {
sol[i] = (j - i) as i32;
break;
} else if sol[j] == 0 {
break;
}
j += sol[j] as usize;
}
}
sol
}
}