数组旋转

2019-02-18 15:32:19 浏览数 (1)

题目:返回将一维数组向右旋转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复制
这里写代码片

0 人点赞