旋转数组简单

藏猫虎

已于 2022-04-20 12:51:27 修改

阅读量106

点赞数

文章标签： c++

于 2022-04-19 22:40:57 首次发布

本文链接：https://blog.csdn.net/BeiWoFW/article/details/124285380

版权

博客介绍了如何在C++中实现将数组向右轮转指定位置的四种方法：原地交换、新建数组、动态数组调整和直接交换。每种方法的时间复杂度和空间复杂度都有所不同，从O(n^2)到O(n)不等，提供了多种思路来解决这个问题。

摘要由CSDN通过智能技术生成

算法二十天

第十天

旋转数组
给你一个数组，将数组中的元素向右轮转 k 个位置，其中 k 是非负数。

示例 1:

输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4] 解释:
向右轮转 1 步:[7,1,2,3,4,5,6]
向右轮转 2 步: [6,7,1,2,3,4,5]
向右轮转 3 步: [5,6,7,1,2,3,4]

示例 2:

输入：nums = [-1,-100,3,99], k = 2
输出：[3,99,-1,-100]
解释: 向右轮转 1 步:[99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

提示：

1 <= nums.length <= 10^5
-231 <= nums[i] <= 231 - 1
0 <= k <= 105

进阶：

尽可能想出更多的解决方案，至少有三种不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的原地算法解决这个问题吗？

作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2skh7/
法一：将数组中的数字一个一个交换到合适位置，时间复杂度O(n^2),空间复杂度O(1)。

#include<bits/stdc++.h>
using namespace std; 
void rotate(vector<int>& nums, int k) {
	int first=nums.size()-(k%nums.size());//移动后数组是以这个数字作为新数组的开头
	for(int i=first;i<nums.size();i++)
	{
		for(int j=i;j>i-first;j--)//将first后的数组向前移动对应位置
			{
				int temp=nums[j];
				nums[j]=nums[j-1];
				nums[j-1]=temp;
			}
	}
}

int main()
{
	vector<int> prices={1,2,3,4,5,6,7};
	rotate(prices,3);
	for(auto& i:prices)
		cout<<i<<"";
}

法二：建立新的数组，遍历一遍即可时间复杂度O(n),空间复杂度O(n)。

#include<bits/stdc++.h>
using namespace std; 
void rotate(vector<int>& nums, int k) {
	int first=nums.size()-(k%nums.size());//同上
	if(first==0)
		return;
	vector<int> temp;
	for(int i=first;i<nums.size();i++)
		temp.push_back(nums[i]);
	for(int i=0;i<first;i++)
		temp.push_back(nums[i]);
	nums=temp;
}

int main()
{
	vector<int> prices={1,2,3,4,5,6,7};
	rotate(prices,3);
	for(auto& i:prices)
		cout<<i<<"";
}

法三：动态数组解决，时间复杂度O(n^2),空间复杂度O(1);

#include<bits/stdc++.h>
using namespace std; 
void rotate(vector<int>& nums, int k) {
	for(int i=0;i<k;i++)
	{
		nums.insert(nums.begin(),nums[nums.size()-1]);//临时的空间损耗不计入总的空间复杂度；
		nums.pop_back();
	}
}

int main()
{
	vector<int> prices={1,2,3,4,5,6,7,8};
	rotate(prices,3);
	for(auto& i:prices)
		cout<<i<<"";
}

法四：找到位置直接交换，时间复杂度O(n),空间复杂度O(1);

#include<bits/stdc++.h>
using namespace std; 
void rotate(vector<int>& nums, int k) {
	int temp1;
	k%=nums.size();
	if(k==0||nums.size()==1)
		return;
	int temp=nums[0];
	int count=0;
	int sum=k;
	for(int j=0;j<nums.size();j++)
		{
			temp1=nums[sum%nums.size()];
			nums[sum%nums.size()]=temp;
			temp=temp1;
			
			if(sum%nums.size()==count)
			{
				count++;
				sum=count;
				temp=nums[sum%nums.size()];
			}
			sum+=k;
		}
}

int main()
{
	vector<int> prices={1,2,3,4,5,6};
	rotate(prices,4);
	for(auto& i:prices)
		cout<<i<<" ";
}