题目连接:
第一想法:
只想到暴力法,之后看完思路自己尝试完成。
思路:
因为是有序序列,并且含有负数,所以平方后从大到小肯定是从序列的两边往中间靠拢。设置两个指针分别指向第一个和最后一个,比较他们大小,放到额外数组。放过的指针++,继续对比,直到全部放完。
视频链接:
双指针法经典题目 | LeetCode:977.有序数组的平方_哔哩哔哩_bilibili
我的解法:
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> result(nums.size(),0);
int len = nums.size() - 1;
int left = 0;
int right = len;
while(left != right)
{
if(nums[left]*nums[left] >= nums[right]*nums[right])
{
result[len--] = nums[left]*nums[left];
left++;
}
else
{
result[len--] = nums[right]*nums[right];
right--;
}
}
result[len]=nums[left]*nums[left];
return result;
}
};
总结:
无法很好掌握循环,使用了指针是否相等作为结束条件,导致最后需要单独拷贝一次。
随想录解法:
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
int k = A.size() - 1;
vector<int> result(A.size(), 0);
for (int i = 0, j = A.size() - 1; i <= j;) { // 注意这里要i <= j,因为最后要处理两个元素
if (A[i] * A[i] < A[j] * A[j]) {
result[k--] = A[j] * A[j];
j--;
}
else {
result[k--] = A[i] * A[i];
i++;
}
}
return result;
}
};