科目一刷题03 有序数组的平方

zhranjie

已于 2024-07-06 11:33:02 修改

阅读量382

点赞数 6

分类专栏：刷题日记文章标签：算法数据结构

于 2024-06-10 21:43:40 首次发布

本文链接：https://blog.csdn.net/zhranjie/article/details/139582791

版权

刷题日记专栏收录该内容

6 篇文章 0 订阅

订阅专栏

977. 有序数组的平方 - 力扣（LeetCode）

双指针法

    int square(int num)
    {
        return num*num;
    }
    vector<int> sortedSquares(vector<int>& nums)
    {
        vector<int> ret;
        int leftIndex = 0;
        int rightIndex = nums.size()-1;
        while(leftIndex<=rightIndex){  // 为什么要==
            if (square(nums[leftIndex])>=square(nums[rightIndex])){
                ret.push_back(square(nums[leftIndex]));
                leftIndex++;    
            }else{
                ret.push_back(square(nums[rightIndex]));
                rightIndex--;
            }
        }
        std::reverse(ret.begin(),ret.end());
        return ret;
    }