fix subtract with overflow

This commit is contained in:
flydzy
2023-02-10 18:40:24 +08:00
committed by GitHub
parent 6884c44f72
commit 4b8b726731

View File

@ -183,14 +183,14 @@ Rust
impl Solution {
pub fn sorted_squares(nums: Vec<i32>) -> Vec<i32> {
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;