问题描述:把一个数组中的元素循环右移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]+",");
}
}