55: Re-submit

sankichi92 · Dec 29, 2023 · 9e96fd6 · 9e96fd6
1 parent 0a2d1a6
commit 9e96fd6
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/55-find-peak-element/src/lib.rs b/55-find-peak-element/src/lib.rs
@@ -2,12 +2,11 @@ pub fn find_peak_element(mut nums: Vec<i32>) -> i32 {
     let length = nums.len();
     nums.insert(0, i32::MIN);
     nums.push(i32::MIN);
-    nums
-        .iter()
+    nums.iter()
         .skip(1)
         .take(length)
         .enumerate()
-        .find(|(i, num)| nums[*i] < **num && **num > nums[*i + 2])
+        .find(|(i, num)| nums[*i] <= **num && **num >= nums[*i + 2])
         .unwrap()
         .0 as i32
 }
@@ -25,4 +24,9 @@ mod tests {
     fn case2() {
         assert_eq!(find_peak_element(vec![1, 2, 1, 3, 5, 6, 4]), 1)
     }
+
+    #[test]
+    fn failed_case1() {
+        assert_eq!(find_peak_element(vec![-2147483648]), 0)
+    }
 }