Q:把一个含有 N 个元素的数组循环右移 K 位,要求时间复杂度为 O (N),且只允许使用两个附加变量。
开始的思路:
1.若k=整数倍N, 完成;k大于N,k=N%k;k小于N,开始处理。
2.tmp1保存取出的元素a[k];a[0]放在k;tmp2保存a[2k], tmp1放入2k;tmp1保存a[3k],tmp2放入3k...直到位置[0]被放入一个新值,一轮循环完毕。这是一个从位置[0]出发,又回到位置[0]的接龙游戏。
仔细考虑下,发现,若N%k==0,则需要从[0]开始,[0]结束...到[k-1]开始[k-1]结束,都执行一遍。若N%k!=0,则可能一次执行完,也可能执行完了0~k-1中的部分,
所以这个方法还需要记录从[0]到[k-1]哪些已经被成功回置新元素,所有都回置过,就完成。
这个方法貌似不是很简练,代码如下:
然后有一个新的方法,跟reverse数组有关:整个过程把数组分为0~n-k-1和n-k~n-1两个部分,那么对左边翻转,右边翻转,再整体翻转,就有最后循环右移k的效果。
用i在左边范围,n-k+i在右边范围都试过,确实最后移到了i+k, i,满足条件
// 将buffer中start和end之间的元素逆序 void Reverse( int buffer[], int start, int end ) { while ( start < end ) { int temp = buffer[ start ] ; buffer[ start++ ] = buffer[ end ] ; buffer[ end-- ] = temp ; } } // 将含有n个元素的数组buffer右移k位 void Shift( int buffer[], int n, int k ) { k %= n ; Reverse( buffer, 0, n - k - 1) ; Reverse( buffer, n - k, n - 1 ) ; Reverse( buffer, 0, n - 1 ) ; }
时间: 2024-12-10 11:45:02