数组旋转
问题:
返回将一维数组向右旋转K个位置的结果。比如,一维数组{1,2,3,4,5},k=2时,返回结果是{4,5,1,2,3}。要求常数级空间复杂度,允许修改原有数组
看到一个比较巧妙的方法,将数组反转三次,第一次反转整个数组,第二次反转数组的前K个数,第三次反转数组剩下的数。
每次反转的时间为O(N)。
例如,
一维数组{1,2,3,4,5},k=2
第一次反转:5,4,3,2,1
第二次反转:4,5,3,2,1
第三次反转:4,5,1,2,3 即为最终结果
代码入下:
int[] rotateK(int[] A,int k){ if(A==null || k>=A.length) return A; reverse(A,0,A.length-1); //反转整个数组 reverse(A,0,A.length-1); //反转前K个数 reverse(A,0,A.length-1); //反转剩下的数 return A; } void reverse(int[] A,int start,int end){ while(start<end){ int temp = A[start]; A[start] = A[end]; A[start] = temp; start++; end--; } }