【代码随想录】算法训练营 第二天 第一章 数组 Part 2

977. 有序数组的平方

题目

暴力解法

思路

原地更新所有数组元素为其平方数后,再使用sort函数排序,对vector使用sort函数时,两个参数分别是vector的起始元素和终止元素。

代码

cpp 复制代码
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        for (int i = 0; i < nums.size(); i++) 
            nums[i] = nums[i] * nums[i];
        sort(nums.begin(), nums.end());
        return nums;
    }
};

双指针法

思路

从题目中我们可以知道,数组元素有正有负,从小到大排序,但是平方以后,元素大小的分布就变成了两边大中间小;

基于此,我们使用双指针法来代替sort函数,双指针在归并排序里就用到了,但在这里,两个指针不是指着两个数组,而是指着输入数组的两端,每次比较两端元素的大小,将较大元素放在新数组里(新数组用vector定义一个和原数组内容一样的,新数组的指针每次向左移动一位),然后指针向内移动一位,直到两个指针相遇。

代码

cpp 复制代码
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int i = 0, j = nums.size() - 1, k = nums.size() - 1;
        vector<int> num = nums;
        while (i <= j) {
            if(fabs(nums[i]) >= fabs(nums[j]))
                num[k--] = nums[i] * nums[i++];
            else
                num[k--] = nums[j] * nums[j--];
        }
        return num;
    }
};

209. 长度最小的子数组

相关推荐
通信小呆呆6 分钟前
以矩阵视角统一理解:外积、Kronecker 积与 Khatri–Rao 积(含MATLAB可视化)
线性代数·算法·matlab·矩阵·信号处理
CoderCodingNo1 小时前
【GESP】C++四级真题 luogu-B4068 [GESP202412 四级] Recamán
开发语言·c++·算法
一个不知名程序员www1 小时前
算法学习入门---双指针(C++)
c++·算法
Shilong Wang2 小时前
MLE, MAP, Full Bayes
人工智能·算法·机器学习
Theodore_10222 小时前
机器学习(6)特征工程与多项式回归
深度学习·算法·机器学习·数据分析·多项式回归
知花实央l2 小时前
【算法与数据结构】拓扑排序实战(栈+邻接表+环判断,附可运行代码)
数据结构·算法
lingling0092 小时前
机械臂动作捕捉系统选型指南:从需求到方案,NOKOV 度量光学动捕成优选
人工智能·算法
吃着火锅x唱着歌2 小时前
LeetCode 410.分割数组的最大值
数据结构·算法·leetcode
Benny_Tang3 小时前
题解:P7989 [USACO21DEC] Bracelet Crossings G
c++·算法
YSRM3 小时前
Leetcode+Java+图论+并查集
算法·leetcode·图论