当前位置: 代码迷 >> 综合 >> leetcode 977. Squares of a Sorted Array 求一个数组的平方
  详细解决方案

leetcode 977. Squares of a Sorted Array 求一个数组的平方

热度:34   发布时间:2023-12-15 14:55:10.0

https://leetcode.com/problems/squares-of-a-sorted-array/

两个指针i,j,分别指向头和尾,平方后比较:

若相等,则将元素的平方压入数组result,左指针i右移;

反之,将较大的元素平方压入数组result,较大的指针向另一个指针移动;

压入最后一个元素,将result数组取反。

class Solution {
public:vector<int> sortedSquares(vector<int>& A) {vector<int> result;if(A.size()==0)return result;if(A.size()==1){result.push_back(A[0]*A[0]);return result;}int i = 0, j = A.size()-1;while(i<j){while(i<j && A[i]*A[i]==A[j]*A[j]){result.push_back(A[i]*A[i]);//如果要求平方后不能有重复值,删除这一句即可i++;}if(i<j && A[i]*A[i]<A[j]*A[j]){result.push_back(A[j]*A[j]);j--;}else if(i<j && A[i]*A[i]>=A[j]*A[j]){result.push_back(A[i]*A[i]);i++;}}result.push_back(A[i]*A[i]);reverse(result.begin(), result.end());return result;}
};

 

 

 

  相关解决方案