From 4bd8f3ea32bb189de5919745e531a964b5fcd008 Mon Sep 17 00:00:00 2001
From: fwqaaq <fwqaaq@gmail.com>
Date: Mon, 21 Aug 2023 16:07:53 +0800
Subject: [PATCH] =?UTF-8?q?Update=200496.=E4=B8=8B=E4=B8=80=E4=B8=AA?=
 =?UTF-8?q?=E6=9B=B4=E5=A4=A7=E5=85=83=E7=B4=A0I.md=20=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=20rust?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 problems/0496.下一个更大元素I.md | 32 +++++++++++++++----------
 1 file changed, 20 insertions(+), 12 deletions(-)

diff --git a/problems/0496.下一个更大元素I.md b/problems/0496.下一个更大元素I.md
index 6bcafafb..e02cfbd1 100644
--- a/problems/0496.下一个更大元素I.md
+++ b/problems/0496.下一个更大元素I.md
@@ -392,25 +392,33 @@ function nextGreaterElement(nums1: number[], nums2: number[]): number[] {
 ### Rust
 
 ```rust
+use std::collections::HashMap;
 impl Solution {
     pub fn next_greater_element(nums1: Vec<i32>, nums2: Vec<i32>) -> Vec<i32> {
-        let mut ans = vec![-1; nums1.len()];
-        use std::collections::HashMap;
-        let mut map = HashMap::new();
-        for (idx, &i) in nums1.iter().enumerate() {
-            map.insert(i, idx);
+        let (mut res, mut map) = (vec![-1; nums1.len()], HashMap::new());
+        if nums1.is_empty() {
+            return res;
         }
+
+        nums1.into_iter().enumerate().for_each(|(v, k)| {
+            map.insert(k, v);
+        });
+
         let mut stack = vec![];
-        for (idx, &i) in nums2.iter().enumerate() {
-            while !stack.is_empty() && nums2[*stack.last().unwrap()] < i {
-                let pos = stack.pop().unwrap();
-                if let Some(&jdx) = map.get(&nums2[pos]) {
-                    ans[jdx] = i;
+        for (i, &value) in nums2.iter().enumerate() {
+            while let Some(&top) = stack.last() {
+                if value <= nums2[top] {
+                    break;
+                }
+                let stacked_index = stack.pop().unwrap();
+                if let Some(&mapped_index) = map.get(&nums2[stacked_index]) {
+                    res[mapped_index] = value;
                 }
             }
-            stack.push(idx);
+            stack.push(i);
         }
-        ans
+
+        res
     }
 }
 ```