// class Solution {
// public:
// vector<int> sortedSquares(vector<int>& A) {
// vector<int> result;
// for(int i = 0, j = A.size()-1; i <= j;){
// result.insert(result.begin(), abs(A[i]) >= abs(A[j]) ? pow(A[i++], 2) : pow(A[j--], 2));
// }
// return result;
// }
// };
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
vector<int> result(A.size());//这个功能经常用到
for(int i = 0, j = A.size()-1, p = A.size()-1; p >= 0; --p){
result[p] = pow(abs(A[i]) >= abs(A[j]) ? A[i++] : A[j--], 2);
}
return result;
}
};
首先,前者的insert语句,每次插入都要移动后面的,很费时间,不如一开始就指定result的大小,直接使用索引填入数字就快很多了。其次,把判断放在pow函数的里面,看起来简洁了很多。最后,p的边界条件一定是要包括=的。这个循环语句一次性初始化了三个变量。
改进后:
Runtime: 100 ms, faster than 98.95% of C++ online submissions for Squares of a Sorted Array.
Memory Usage: 13.4 MB, less than 100.00% of C++ online submissions for Squares of a Sorted Array.