アルゴリズム20日
10日目
配列の回転 配列が
与えられた場合、配列内の要素を 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
高度な:
できるだけ多くの解決策を考えてください。この問題に対処するには少なくとも 3 つの異なる方法があります。
O(1) 空間複雑さのインプレース アルゴリズムを使用して、この問題を解決できますか?
著者: LeetCode
リンク: https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2skh7/
方法1: 配列内の数値を1つずつ適切な位置に置き換える、時間が複雑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<<"";
}
方法 2: 新しい配列を作成し、それを 1 回走査する 時間計算量は 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<<"";
}
方法 3: 動的配列ソリューション、時間計算量 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<<"";
}
方法 4: 位置を見つけて直接交換します。時間計算量 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<<" ";
}