给定一个整数数组 nums
,将数组中的元素向右轮转 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]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/rotate-array
思路1:
开一个新的数组存放交换两段的数组,最后再遍历
void rotate(int* nums, int numsSize, int k){
k = k%numsSize;
int j = 0;
int tmp[numsSize];
for(int i = numsSize - k;i<numsSize;i++)
{
tmp[j]=nums[i];
j++;
}
for(int i = 0;i<numsSize-k;i++)
{
tmp[j]=nums[i];
j++;
}
for(int i = 0;i<numsSize;i++)
{
nums[i]=tmp[i];
}
return nums;
}
思路2:
前n-k个逆置,后k个逆置,整体逆置
void reverse(int* a,int begin,int end)
{
while(begin<end)
{
int tmp = a[begin];
a[begin]=a[end];
a[end]=tmp;
begin++;
end--;
}
}
void rotate(int* nums, int numsSize, int k)
{
k%=numsSize;
reverse(nums,0,numsSize-k-1);
reverse(nums,numsSize-k,numsSize-1);
reverse(nums,0,numsSize-1);
}