有序数组的平方 - C++
题目(LeetCode 977)
给定一个按非递减顺序排序的整数数组 A,返回每个数字的平方组成的新数组,要求也按非递减顺序排序。
(1) 1 <= A.length() <= 10000
(2) -10000 <= A[i] <= 10000
(3) A 已按非递减顺序排序。
示例1
输入: [-4,-1,0,3,10]
输出: [0,1,9,16,100]示例2
输入: [-7,-3,2,3,11]
输出: [4,9,9,49,121]
方法一:排序
一种简单的思路是依次求出A
中所有数字的平方,然后对该数组按非递减顺序排序。代码如下。
vector<int> sortedSquares(vector<int>& A) {
vector<int> ans(A.size(), 0);
for (int i = 0; i < A.size(); i++)
ans[i] = A[i] * A[i];
sort(ans.begin(), ans.end());
return ans;
}
我们对整个数组遍历了一次,时间复杂度是 O ( n ) O(n) O(n),使用sort()
函数排序的时间复杂度是 O ( n log n ) O(n\log n)