编程之美2.17 数组循环移位

问题描述

把一个含有N个元素的数组循环右移K位，如K=4的时候abcd1234->1234abcd。一般考虑K>0的情况，即右移；同时K<N，当然K>=N的话可以通过K%=N得到一样的结果

 

解法思路详见代码注释

/****************************问题描述*****************************把一个含有N个元素的数组循环右移K位，如K=4的时候abcd1234->1234abcd一般考虑K>0的情况，即右移；同时K<N，当然K>=N的话可以通过K%=N得到一样的结果******************************************************************/#include<iostream>#include<cstring>using namespace std;/****************************解法一：暴力法*************************每次移动一位，共移动K次，假设K=4，则有abcd1234->4abcd123->34abcd12->234abcd1->1234abcd时间复杂度为O(N*N)********************************************************************/void RightShift1(char* arr,int len,int k){k%=len;while(k--){char tmp=arr[len-1];for(int i=len-1;i>0;i--)arr[i]=arr[i-1];arr[0]=tmp;}}/************************解法二：暴力法变种**************************先用一个K位的数组保存数组最后的K个元素，然后对原数组的第N-K位到第1位，每位直接移动到其原来位置的后K位，最后将预先保存的K位复制到最前面，如：结社K=4,则有abcd1234->0000abcd->1234abcd比解法一稍微快一点，但是需要额外的空间存储********************************************************************/void RightShift2(char* arr,int len,int k){k%=len;//生成一个长度为K的临时数组储存最后K位数char* tmp=new char[k];//将arr的后K个元素赋值到tmp里面for(int i=0;i<k;i++)tmp[i]=arr[len-k+i];//将arr的前N-K个元素每一个元素直接往后挪K位for(int i=len-k-1;i>=0;i--)arr[i+k]=arr[i];//将tmp的元素赋值到arr的前K位for(int i=0;i<k;i++)arr[i]=tmp[i];}/***********************解法三：字符翻转****************************将arr的前N-K个元素翻转，再将后K个元素翻转，最后将整个数组翻转，例如1234abcd->4321abcd->4321dcba->abcd1234********************************************************************/void Reverse(char* arr,int begin,int end){for(;begin<end;begin++,end--){char tmp=arr[begin];arr[begin]=arr[end];arr[end]=tmp;}}void RightShift3(char* arr,int len,int k){Reverse(arr,0,len-k-1);Reverse(arr,len-k,len-1);Reverse(arr,0,len-1);}#define MAIN#ifdef MAINint main(){char a[]="1234abcd";int k=4; //右移位数cout<<"Before RightShift:"<<endl;cout<<a<<endl;cout<<"After RightShift:"<<endl;RightShift3(a,strlen(a),k);cout<<a<<endl;system("PAUSE");return 0;}#endif