题目:返回将一维数组向右旋转k个位置的结果。比如,一维数组{1,2,3,4,5},当k=2时,返回结果是{4,5,1,2,3}。要求常数级空间复杂度,允许修改原有数组。
伪代码
代码语言:javascript复制int [] rotataK(int [] A,int k)
{
if (A == null || k >= A.length) return A;
reverse(A,0,A.length-1);//反转整个数组
reverse(A,0,k-1);//反转前k个数
reverse(A,k,A.length-1);//反转剩下的数
return A;
}
//辅助函数,反转从start到end的数
void reverse (int [] A,int start,int end)
{
while (start < end)
{
//交换A[start]和A[end]两个数
int temp = A[start]
A[start] = A[end]
start ;
end--;
}
}
R语言
R语言实现这个功能上,采用两种方式:
- R语言自带的rev函数
- 自己写的翻转函数
R语言带的rev函数
代码语言:javascript复制rotateK <- function(a,k)
{
if (is.null(a) || length(a) <= k)
return(a)
a <- rev(a)
b <- rev(a[1:k])
b1 <- rev(a[(k 1):length(a)])
b <- append(b,b1)
return (b)
}
> a<-c(1,2,3,4,5)
> a
[1] 1 2 3 4 5
> k=2
> rotateK(a,k)
[1] 4 5 1 2 3
自定义的翻转函数
代码语言:javascript复制fanzhuan <- function(a,start,end)
{
while(start < end)
{
temp = a[start]
a[start] = a[end]
a[end] = temp
start = start 1
end = end - 1
}
return(a)
}
rotateK_1 <- function(a,k)
{
if (is.null(a) || length(a) <= k)
{
return(a)
}
a <- fanzhuan(a,1,length(a))
b <- fanzhuan(a,1,k)
b1 <- fanzhuan(a,(k 1),length(a))
b <- append(b[1:k],b1[(k 1):length(a)])
return (b)
}
> a<-c(1,2,3,4,5)
> fanzhuan(a,1,5)
[1] 5 4 3 2 1
> rev(a)
[1] 5 4 3 2 1
两种方法的比较
k =2
代码语言:javascript复制> #k=2
> a<-c(1,2,3,4,5)
> k=2
> rotateK(a,k)
[1] 4 5 1 2 3
> #k=2
> k=2
> a<-c(1,2,3,4,5)
> rotateK_1(a,k)
[1] 4 5 1 2 3
k =3
代码语言:javascript复制> #k=3
> a<-c(1,2,3,4,5)
> k=3
> rotateK(a,k)
[1] 3 4 5 1 2
> #k=3
> k=3
> a<-c(1,2,3,4,5)
> rotateK_1(a,k)
[1] 3 4 5 1 2
k = 4
代码语言:javascript复制> #k=4
> a<-c(1,2,3,4,5)
> k=4
> rotateK(a,k)
[1] 2 3 4 5 1
> #k=4
> k=4
> a<-c(1,2,3,4,5)
> rotateK_1(a,k)
[1] 2 3 4 5 1
就这样解密了R语言中的rev源代码啦!~~~~
python
代码语言:javascript复制这里写代码片