编程之美2.17—数组循环移位(旋转数组)

来源:互联网 发布:剑三军爷男神捏脸数据 编辑:程序博客网 时间:2024/05/29 08:03

题目:

把一个含有N个元素的额数组循环右移K位,要求时间复杂度O(N),且只允许使用两个附加变量。


解法一:O(N^2)

每次将数组中的元素右移移位,循环K次。当K>N时,右移K位和右移K%N位是一样的。

MyShift(int a[],int N,int K){     K%=N;     while(K--)     {         int t=a[N-1];         for(int i=N-1;i>0;i--)             a[i]=a[i-1];         a[0]=t;      }}

解法二:O(N)

假如数组abcd1234

1.逆序abcd:abcd1234——>dcba1234;

2.逆序1234:dcba1234——>dcba4321;

3.逆序全部:dcba4321——>1234abcd;

MyReverse(int a[],int b,int e){    for(;b<e;b++,e--)    {        swap(a[b],a[e]);     }}MyShift(int a[],int N,int K){    K%=N;    MyReverse(a,0,N-K-1);    MyReverse(a,N-K,N-1);    MyReverse(a,0,N-1);}


0 0
原创粉丝点击