题目描述:设计一个算法,把一个含有N个元素的数组循环右移K位,要求时间复杂度为O(N),且只允许使用两个附加变量,如abcd1234循环右移三位,
则为 abcd11234 ->4abcd123->34abcd12->234abcd1
思路一:暴力解法,一步步来,每次循环移位一次,移K%N次,此时复杂度为O(N^2)。
思路二:比较巧妙的思路,abcd1234->234abcd1,不难看出,其中两段:234和abcd1顺序是不变的,把这两段看成是两个整体。右移K位的过程中就是把数组的两部分交换一
下。变换的过程中通过以下步骤完成:
1.逆序排序:abcd1234->abcd1432
2.逆序排列:abcd1432->1dcba432
3.全部逆序:1dcba432->234abcd1
具体代码如下:
#include <iostream>
#include <stdlib.h>
#include <vector>
using namespace std;
void swap(char& a, char& b)
{
char t = a;
a = b;
b = t;
}
void reverse(vector<char>& A, int left, int right)
{
int i,j;
for(i = left, j = right; i < j; i++,j--)
{
swap(A[i],A[j]);
}
}
void rightShift(vector<char>& A, int k)
{
if(A.size() == 0)
return ;
k = k%A.size();
reverse(A,0,A.size() - k - 1);
reverse(A,A.size() - k, A.size() - 1);
reverse(A,0,A.size() - 1);
}
int main()
{
char a[] = {'1','2','3', '4', 'a','b','c','d'};
vector<char> temp(a, a+ 8);
for(vector<char>::iterator iter = temp.begin(); iter != temp.end(); ++iter)
cout << *iter << " " ;
cout << endl;
rightShift(temp,3);
for(vector<char>::iterator iter = temp.begin(); iter != temp.end(); ++iter)
cout << *iter << " " ;
cout << endl;
system("pause");
}