問題描述:把一個數組中的元素循環右移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]+",");
}
}