题目来源:链接: [https://leetcode-cn.com/problems/squares-of-a-sorted-array/comments/].
977. 有序数组的平方
1.问题描述
给定一个按非递减顺序排序的整数数组 A,返回每个数字的平方组成的新数组,要求也按非递减顺序排序。
示例 1:
输入:[-4,-1,0,3,10]
输出:[0,1,9,16,100]
示例 2:
输入:[-7,-3,2,3,11]
输出:[4,9,9,49,121]
提示:
1. 1<= A.length <= 10000
2. -10000 <= A[i] <= 10000
3. A 已按非递减顺序排序。
2.解决方案
直接数组下标解决
由于A已经按非递减顺序排序,难度降低一个档次
刚开始我的思考是用迭代器去访问容器的每个元素,逐个弹出求平方,然后push进容器,但是A.pop_back() 的返回值是 void ,就没继续往下想
第一遍的代码如下
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
int tmp = 0;
vector<int>::iterator it = A.begin();
for(it = A.begin(); it != A.end(); it++)
{
tmp = A.pop_back(); //这地方报错 因为pop_back没有返回值
tmp *= tmp;
A.push_back(tmp);
}
sort(A.begin(),A.end());
return A;
}
};
3.代码
最终提交的代码如下:
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
int num = A.size();
int i = 0;
for(i = 0; i < num; i++)
{
A[i] = A[i]*A[i];
}
sort(A.begin(),A.end());
return A;
}
};
执行用时: 248 ms, 在Squares of a Sorted Array的C++提交中击败了8.47% 的用户
内存消耗: 11 MB, 在Squares of a Sorted Array的C++提交中击败了100.00% 的用户
2019/2/23 胡云层 于南京 2