轮转数组
- 题目
- 第一种解法:额外数组
- 第二种解法:环状替换
- 第三种解法:翻转数组
- 结语
题目
题目链接:轮转数组
给定一个整数数组 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]
提示:
1 <= nums.length <= 105
-231 <= nums[i] <= 231 - 1
0 <= k <= 105
进阶:
尽可能想出更多的解决方案,至少有 三种 不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
第一种解法:额外数组
代码如下:
void rotate(int* nums, int numsSize, int k) {
int newArr[numsSize];
for (int i = 0; i < numsSize; ++i) {
newArr[(i + k) % numsSize] = nums[i];
}
for (int i = 0; i < numsSize; ++i) {
nums[i] = newArr[i];
}
}
复杂度分析:
时间复杂度: O(n),其中 n 为数组的长度。
空间复杂度: O(n)。
个人分析:
这种解法最妙的地方是(i+k)%numSize的运用,值得多想想,这里是使用了一个变长数组的写法,使用了额外的数组空间接收反转后的数组,再重新赋回原数组。
举个例子:
nums[ ]={1,2,3,4,5},k=2,反转后应是nums[ ]={4,5,1,2,3};
初始条件:i=0 k=2 numsSize=5
第一趟 i=0 (i + k) % numsSize=2 newArr[2]=nums[0]
newArr[]={ , ,1, , }
第二趟 i=1 (i + k) % numsSize=3 newArr[3]=nums[1]
newArr[]={ , ,1,2, }
第三趟 i=2 (i + k) % numsSize=4 newArr[4]=nums[2]
newArr[]={ , ,1,2,3}
第三趟 i=3 (i + k) % numsSize=0 newArr[0]=nums[3]
newArr[]={4, ,1,2,3}
第四趟 i=4 (i + k) % numsSize=1 newArr[1]=nums[4]
newArr[]={4,5,1,2,3}
i=numsSize 终止第一个循环
最后将利用for循环将newArr[ ]中的元素遍历赋给nums数组即可。
第二种解法:环状替换
代码如下:
//最大公约数
int gcd(int a, int b) {
return b ? gcd(b, a % b) : a;
}
//交换
void swap(int* a, int* b) {
int t = *a;
*a = *b, *b = t;
}
void rotate(int* nums, int numsSize, int k) {
k = k % numsSize;
int count = gcd(k, numsSize);
for (int start = 0; start < count; ++start) {
int current = start;
int prev = nums[start];
do {
int next = (current + k) % numsSize;
swap(&nums[next], &prev);
current = next;
} while (start != current);
}
}
复杂度分析:
时间复杂度:O(n),其中 n 为数组的长度。每个元素只会被遍历一次。
空间复杂度:O(1)。我们只需常数空间存放若干变量。
个人分析:
首先讲讲这里的gcd函数是一个求最大公约数的自定义函数,有很多小伙伴可能不理解这种写法,可以替换成下面这种常规的辗转相除写法:
int gcd(int a, int b) {
while(a%b)
{
int r=a%b;
a=b;
b=r;
}
return b;
}
swap是常用的交换函数就不多讲了,这种写法的主要思路是将间隔为k的元素逐个向后替换,再将该间隔最后一个元素与开头替换,求k与numsSize的最大公约数是为了将不同间隔数分组,设定循环次数,便于遍历交换。
举个例子:
nums[ ]={1,2,3,4,5,6},k=2,反转后应是nums[ ]={5,6,1,2,3,4};
初始条件:start=0 k=2 numsSize=6
第一趟 prev=nums[0]=1
current=0 next=(current + k) % numsSize=2
nums[2]=1 prev=3 nums[]={1,2,1,4,5,6}
current=2 next=(current + k) % numsSize=4
nums[4]=3 prev=5 nums[]={1,2,1,4,3,6}
current=4 next=(current + k) % numsSize=0
nums[0]=5 prev=1 nums[]={5,2,1,4,3,6}
current=0=start 终止do...while循环
第二趟 prev=nums[1]=2
current=1 next=(current + k) % numsSize=3
nums[3]=2 prev=4 nums[]={5,2,1,2,3,6}
current=3 next=(current + k) % numsSize=5
nums[5]=4 prev=6 nums[]={5,2,1,2,3,4}
current=5 next=(current + k) % numsSize=1
nums[1]=6 prev=2 nums[]={5,6,1,2,3,4}
current=1=start 终止do...while循环
start=2=count 终止for循环
提一下这里的k=k%numsSize操作,我觉得是个优化吧,假设这里k=8,实际和k=2的最终结果是一样的,这样可以做性能上的优化吧,然后就是这里count是用于循环条件的,也可以不使用最大公约数的赋值写法,也可以在while循环里加一个变量记录被访问的元素,当count等于这个值时,结束遍历也可。
第三种解法:翻转数组
代码如下:
//交换
void swap(int* a, int* b) {
int t = *a;
*a = *b, *b = t;
}
//数组反转
void reverse(int* nums, int start, int end) {
while (start < end) {
swap(&nums[start], &nums[end]);
start++;
end++;
}
}
void rotate(int* nums, int numsSize, int k) {
k %= numsSize;
reverse(nums, 0, numsSize - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, numsSize - 1);
}
复杂度分析:
时间复杂度:O(n),其中 n 为数组的长度。每个元素被翻转两次,一共 n 个元素,因此总时间复杂度为 O(2n)=O(n)。
空间复杂度:O(1)。
个人分析:
这个算法的思路是先将整个数组先反转,再将前后两部分分别反转,个人认为比前两种写法更好理解一些。这里的反转的写法是采用前后指针的交换写法。同样的,举个例子:
nums[ ]={1,2,3,4,5},k=2,反转后应是nums[ ]={4,5,1,2,3};
初始条件: k=2 numsSize=5
第一趟 nums[]={5,4,3,2,1}
第二趟 nums[]={4,5,3,2,1}
第三趟 nums[]={4,5,1,2,3}
结语
这里的解法代码来自力扣官方,作者只是进行了详细的剖析和部分改动
方便大家理解和提升自己,学会多角度观察问题,解决问题。
有兴趣的小伙伴可以关注作者,如果觉得内容不错,请给个一键三连吧,蟹蟹你哟!!!
制作不易,如有不正之处敬请指出
感谢大家的来访,UU们的观看是我坚持下去的动力
在时间的催化剂下,让我们彼此都成为更优秀的人吧!!!