设计一个算法,把一个含有N个元素的数组循环右移K位。要求时间复杂度为O(N)。
#include <stdio.h>
#include <stdlib.h>
void Reverse(char *arr, int b, int e);
void RightShift(char *arr, int N, int k);
int main(void)
{
char a[]="abcd1234";
RightShift(a,sizeof(a)-1,3);
printf("right shift result:%s\n",a);
return EXIT_SUCCESS;
}
void Reverse(char *arr, int b, int e)
{
for (; b < e; b++, e--)
{
int temp = arr[e];
arr[e] = arr[b];
arr[b] = temp;
}
}
void RightShift(char *arr, int N, int k)
{
k %= N;
Reverse(arr, 0, N - k - 1);
Reverse(arr, N - k, N - 1);
Reverse(arr, 0, N - 1);
}
方法二:更高效,只遍历了一次
/*
============================================================================
旋转 在区间[begin,mid)与[mid,end)互换
方法一:循环左移 类似循环右移:rightShift,即前半段倒置,后半段倒置,全部倒置,OK
方法二:挖洞 填坑,两个指针,一个指向洞,另一个指向要用来填坑的元素
在前半段的某个位置,挖个洞,那后填隔mid-begin的位置的元素,即最后旋转完最后应放的元素
时间复杂度O(n),超快
============================================================================
*/
#include <stdio.h>
#include <stdlib.h>
//辗转相除法,求最大公约数
int gcd(int m,int n)
{
while(n!=0)
{
int t=n;
n=m%n;
m=t;
}
return m;
}
//shift偏移量,这里==mid-begin
void rotate_recyle(int*begin,int*end,int*initial,int shift)
{
int t=*initial;
int len=end-begin;
int *p1=initial; //p1为初始位置
int *p2=p1+shift; //p2指向p1最后要放元素的位置,实际上就是p1+shift
while(p2!=initial)
{
*p1=*p2;
p1=p2;
p2=begin+(p2-begin+shift)%len; //p2超过未尾的话,要从头开始
}
*p1=t;
}
/*上面 p2=begin+(p2-begin+shift)%len也可以写成,仿STL
if(end-p2>shift) p2+=shift;
else p2=begin+(shift-(end-p2));
*/
void rotate(int*begin,int*mid,int*end)
{
if(begin==mid || mid==end) return;
int n=gcd(mid-begin,end-begin);
while(n--)
{
rotate_recyle(begin,end,begin+n,mid-begin);
}
}
void pr_arr(int arr[],int len)
{
int i=0;
for(;i<len;i++) printf("%d ",*arr++);
}
int main(void) {
//printf("%d ",gcd(4,8));
int s[]={1,2,3,4,5,6,7,8};
int len=sizeof(s)/sizeof(*s);
rotate(s,s+3,s+len);
pr_arr(s,len);
return 0;
}