LeetCode练习——977.有序数组的平方（C++）

杨哥学编程

于 2024-10-03 21:01:46 发布

阅读量550

点赞数 16

分类专栏： leetcode刷题文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/m0_73239022/article/details/142684455

版权

leetcode刷题专栏收录该内容

3 篇文章 0 订阅

订阅专栏

LeetCode练习——977.有序数组的平方（C++）

题目链接
题目描述
解题思路
总结

题目链接

有序数组的平方

题目描述

给你一个按非递减顺序排序的整数数组 $n u m s$ ，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

示例 1：

输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]

示例 2：

输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

提示：

$1 <= n u m s . l e n g t h <= 104$
$- 104 <= n u m s [i] <= 104$
$n u m s$ 已按非递减顺序排序

进阶：
请你设计时间复杂度为 O(n) 的算法解决本问题。

解题思路

循环+排序

很容易实现的一个方法，其思想就是对数组中的每一个数平方后放回原数组，然后再对数组进行排序就可以了。
具体实现如下：

class Solution {
	vector<int> sortedSquares1(vector<int>& nums) {
		vector<int> result;
		for (int i = 0; i < nums.size(); i++) {
    		nums[i] = nums[i] * nums[i];
		}
		sort(nums.begin(), nums.end() - 1);

		return nums;
	}
};

时间复杂度： $O (n l o g n)$
$C + +$ 中的 $sor t$ 方法的时间复杂度是 $O (n l o g n)$
此处除了使用 $C + +$ 中的 $sor t$ 方法实现排序外，也可以自己实现排序的过程，常见的排序算法有冒泡排序、选择排序、插入排序、归并排序、快速排序、堆排序等。

双指针

众所周知，一个数的绝对值越大，则它平方之后就越大。题目给出的是一个非递减的数组，由此，平方之后最大的一个数不是在最左边就是最右边，利用这个特点，可以从两端向中间遍历数组，然后将平方后最大的数加入到新数组 $res u lt$ ，这里要注意的是在结果数组中需要从最后一个向前遍历添加元素，这样才能得到题目要求的结果也是非递减的。

代码展示

class Solution {
	vector<int> sortedSquares1(vector<int>& nums) {
		// 定义一个新数组，用来保存结果集
		vector<int> result;
		// 给数组足够的空间，放置发生越界访问
		result.resize(nums.size());
		// 上面两行代码可以用vector<int> result = vector<int>(nums.size(), 0)代替
		
		// 新数组添加元素的位置，从最后一个向前添加
		int index = nums.size() - 1;
		// 从两端向中间遍历数组
		for (int left = 0, right = nums.size() - 1; left <= right; ) {
			// 较大的一个平方数在左边
   			if (nums[left] * nums[left] > nums[right] * nums[right]) {
   				// 给result[index]赋值，然后index-1
        		result[index--] = nums[left] * nums[left];
        		// 向右遍历数组
        		left++;
    		}
    		// 较大的一个平方数在右边
    		else {
    			// 给result[index]赋值，然后index-1
        		result[index--] = nums[right] * nums[right];
        		// 向左遍历数组
        		right--;
    		}
		}
		return result;
	}
};

时间复杂度： $O (n)$

总结

在双指针中，循环结束的条件一定是 $l e f t <= r i g h t$ ，如果是 $l e f t < r i g h t$ ，当 $l e f t == r i g h t$ 时，循环结束， $n u m s [l e f t]$ （也可以写成 $n u m s [r i g h t]$ ）就不会被访问到，导致结果不符合预期。也就是说，最后我们的结果中将会缺少 $nums[left]^2$ 。