977. 有序数组的平方

题目连接:

977. 有序数组的平方

第一想法:

        只想到暴力法,之后看完思路自己尝试完成。

思路:

        因为是有序序列,并且含有负数,所以平方后从大到小肯定是从序列的两边往中间靠拢。设置两个指针分别指向第一个和最后一个,比较他们大小,放到额外数组。放过的指针++,继续对比,直到全部放完。

视频链接:

双指针法经典题目 | LeetCode:977.有序数组的平方_哔哩哔哩_bilibili

 

我的解法:

class Solution {
public:
    vector sortedSquares(vector& nums) {
        vector 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 sortedSquares(vector& A) {
        int k = A.size() - 1;
        vector 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;
    }
};

你可能感兴趣的:(leetcode,算法)