Merge pull request #1890 from flydzy/master

fix subtract with overflow
This commit is contained in:
程序员Carl
2023-02-12 11:22:56 +08:00
committed by GitHub

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;