From 4b8b7267310e84cfa414bd80379d384abcf89d57 Mon Sep 17 00:00:00 2001 From: flydzy <77653369+flydzy@users.noreply.github.com> Date: Fri, 10 Feb 2023 18:40:24 +0800 Subject: [PATCH] fix subtract with overflow --- problems/0977.有序数组的平方.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/problems/0977.有序数组的平方.md b/problems/0977.有序数组的平方.md index b8890052..996a7cee 100644 --- a/problems/0977.有序数组的平方.md +++ b/problems/0977.有序数组的平方.md @@ -183,14 +183,14 @@ Rust impl Solution { pub fn sorted_squares(nums: Vec) -> Vec { let n = nums.len(); - let (mut i,mut j,mut k) = (0,n - 1,n- 1); + let (mut i,mut j,mut k) = (0,n - 1,n); let mut ans = vec![0;n]; while i <= j{ if nums[i] * nums[i] < nums[j] * nums[j] { - ans[k] = nums[j] * nums[j]; + ans[k-1] = nums[j] * nums[j]; j -= 1; }else{ - ans[k] = nums[i] * nums[i]; + ans[k-1] = nums[i] * nums[i]; i += 1; } k -= 1;