Python对数组进行循环移位
要求
对含有N个元素的数组循环右移K位,要求时间复杂度为O(N),且只允许使用两个附加变量。
分析
方法一:蛮力法
要求将数组元素循环右移K位,只需要每次将数组中元素右移一位,循环K次即可。如原数组为abcd1234,右移4位具体移动过程为abcd1234-->4abcd123-->34abcd12-->1234abcd。
方法二:翻转法
直接上例子,对于数组序列A = [123456],如何实现循环右移2位功能?将数组A分成两个部分A[0,N-K-1]和A[N-K,N-1],将这两部分分别翻转,然后放在一起再翻转,具体如下:
- ①翻转1234:123456-->432156
- ②翻转56:432156-->432165
- ③翻转432165:432165-->561234
代码实现
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | #方法一 # -*- coding:utf-8 -*- def rightShift(arr,k): if arr = = None : print ( "参数不合法!" ) return lens = len (arr) k % = lens #因为K不一定小于N,有可能大于等于N,当K≥N时,右移K-N与右移K位效果一样 while k ! = 0 : #右移k位 tmp = arr[lens - 1 ] #数组最后一个元素放入临时变量中 i = lens - 1 while i > 0 : arr[i] = arr[i - 1 ] #所有元素后移 i - = 1 arr[ 0 ] = tmp #第一个元素为初始最后一个元素的值 k - = 1 if __name__ = = "__main__" : k = 4 arr = [ 'a' , 'b' , 'c' , 'd' , '1' , '2' , '3' , '4' ] rightShift(arr,k) i = 0 while i < len (arr): print (arr[i],end = "") i + = 1 |
运行结果:
1234abcd
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | #方法二 def reverse(arr,start,end): while start<end: temp = arr[start] arr[start] = arr[end] arr[end] = temp start + = 1 end - = 1 def rightShift(arr,k): if arr = = None : print ( "参数不合法!" ) return lens = len (arr) k % = lens reverse(arr, 0 ,lens - k - 1 ) reverse(arr,lens - k,lens - 1 ) reverse(arr, 0 ,lens - 1 ) if __name__ = = "__main__" : k = 4 arr = [ 'a' , 'b' , 'c' , 'd' , '1' , '2' , '3' , '4' ] rightShift(arr,k) i = 0 while i < len (arr): print (arr[i],end = "") i + = 1 |
运行结果
1234abcd
性能分析
方法一每移动一次,其时间复杂度为O(N),故移动K次,总的时间复杂度为O(K*N),0<K<N,且时间复杂度不满足O(N)。
方法二时间复杂度为O(N),完成翻转操作只用了一个辅助存储空间。
以上综合自网络,希望能给大家一个参考!
转自:微点阅读 https://www.weidianyuedu.com