给你一个按 非递减顺序 排序的整数数组 nums
,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
解法一:暴力求解
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> list;
for (int num : nums) {
list.push_back(num * num);
}
sort(list.begin(), list.end());
return list;
}
};
解法二:双指针
数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> list(nums.size(), 0);
int k = nums.size() - 1; // 新数组的大小为k
int i = 0, j = nums.size() - 1; // 两个指针,首尾
while (i <= j) {
if (nums[i] * nums[i] < nums[j] * nums[j]) {
list[k--] = nums[j] * nums[j];
j--;
}
else {
list[k--] = nums[i] * nums[i];
i++;
}
}
return list;
}
};