问题描述:把一个数组中的元素循环右移k位,时间复杂度严格为O(n),不能是O(kn).
分析:对于这个问题很容易想到的一种方法是依次循环右移,但是这样的话时间复杂度是O(kn),明显不符合题目要求,在之前的博客中,我写的对于字符串的移位问题,可以借助里面的方法三步反转法。
第一步:根据n和k求出分界点的下标,假设为index,则index=n-k(数组下标从0开始),将整个数组分为两部分,0~index-1和index~n-1,
第二步:将0~index-1部分反转。
第三布:将index~n-1部分反转。
第四步:将整个数组反转。
具体的Java实现代码如下:
public class Main { public static void removerightk(int a[],int k){ if(k<0)return; //如果k小于0,则退出 k=k%a.length; //由于k可能大于n,所以将k化为1~n之间 int index=a.length-k; reverse(a,0,index-1); reverse(a,index,a.length-1); reverse(a,0,a.length-1); } public static void reverse(int a[],int s,int e){ while(s<e){ int t=a[s]; a[s]=a[e]; a[e]=t; s++; e--; } } public static void main(String[] args) { int a[]={0,1,2,3,4,5,6,7,8,9}; removerightk(a, 2); for(int i=0;i<a.length;i++) System.out.print(a[i]+","); } }