977.有序数组的平方

标题977.有序数组的平方

题目描述

题目链接

方法1：暴力解题

#pragma once
#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;

//方法1：思路-将容器中的数平方后直接排序
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums)
    {
        int n = nums.size();
        cout << "原始数组" << endl;

        for (int i = 0; i < n; i++)
        {
            cout << nums[i] << " ";
        }
        cout << endl;

        for (int i = 0; i < n; i++)
        {
            nums[i] = nums[i] * nums[i];
        }

        cout << "排序之前" << endl;
        vector<int>::iterator it;
        for (it = nums.begin(); it != nums.end(); it++)
        {
            cout << *it << " ";
        }
        cout << endl;

        sort(nums.begin(), nums.end());
        return nums;

    }
};

int main()
{

    vector<int> v;
    v.push_back(-4);
    v.push_back(-1);
    v.push_back(0);
    v.push_back(3);
    v.push_back(10);

    Solution S;
    S.sortedSquares(v);
    
	system("pause");
	return 0;
}

运行结果

提交结果

方法2：对撞双指针法

#pragma once
#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;

//方法2：对撞双指针法
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums)
    {
        //新建一个容器用于保存结果 初始化为0
        vector<int> result_v(nums.size(), 0);
        int index = result_v.size() - 1;

        int n = nums.size();
        //创建前后指针
        int frontp = 0;
        int rearp = n - 1;

        while (frontp <= rearp)
        {
            if (nums[frontp] * nums[frontp] < nums[rearp] * nums[rearp])
            {
                //如果前指针对应的值平方小于后指针值的平方，大的值作为结果容器的最后元素
                result_v[index] = nums[rearp] * nums[rearp];
                //结果元素的最后元素索引-1
                index--;
                //后指针前移
                rearp--;
            }
            else
            {
                //如果前指针对应的值平方大于后指针值的平方，大的值作为结果容器的最后元素
                result_v[index] = nums[frontp] * nums[frontp];
                //结果元素的最后元素索引-1
                index--;
                //前指针前移
                frontp++;
            }
        }

        //遍历结果容器
        vector<int>::iterator it;
        for (it = result_v.begin(); it != result_v.end(); it++)
        {
            cout << *it << " ";
        }
        cout << endl;
        //返回结果容器
        return result_v;


    }
};


int main()
{

    vector<int> v;
    v.push_back(-4);
    v.push_back(-1);
    v.push_back(0);
    v.push_back(3);
    v.push_back(10);

    Solution S;
    S.sortedSquares(v);
    
	system("pause");
	return 0;
}

运行结果

提交结果