Update 0503.下一个更大元素II.md

This commit is contained in:
Zhipeng Xu
2023-07-08 20:38:47 +08:00
committed by jinbudaily
parent 4760924db0
commit b3d1a88fde

View File

@ -266,6 +266,24 @@ function nextGreaterElements(nums: number[]): number[] {
};
```
Rust
```rust
impl Solution {
pub fn next_greater_elements(nums: Vec<i32>) -> Vec<i32> {
let mut ans = vec![-1; nums.len() * 2];
let mut stack = vec![];
let double = nums.repeat(2);
for (idx, &i) in double.iter().enumerate() {
while !stack.is_empty() && double[*stack.last().unwrap()] < i {
let pos = stack.pop().unwrap();
ans[pos] = i;
}
stack.push(idx);
}
ans.into_iter().take(nums.len()).collect()
}
}
```
<p align="center">