面试题:旋转数组

最新推荐文章于 2024-01-11 07:00:00 发布

chfe910

最新推荐文章于 2024-01-11 07:00:00 发布

阅读量828

点赞数 1

分类专栏： Algorithms 文章标签：搜狐笔试笔试题旋转数组 2015 翻转

本文链接：https://blog.csdn.net/chfe007/article/details/45081095

版权

Algorithms 专栏收录该内容

22 篇文章 1 订阅

订阅专栏

这是搜狐2015年实习生招聘研发岗位笔试的第一道编程题.

题目描述:输入一个数组,将其循环旋转k个位置,要求时间复杂度为O(N),空间复杂度越低越好.

以一个例子来说明:
初始数组: 1,2,3,4,5
k=2
首先倒置整个数组:5,4,3,2,1
然后倒置前k个:4,5,3,2,1
然后倒置后面剩下的:4,5,1,2,3
以上得到的结果即为所求.

代码实现如下:

void reverse(int *A, int start, int end)
{
    while (start < end)
    {
        int tmp = A[start];
        A[start] = A[end];
        A[end] = tmp;
        ++start;
        --end;
    }
}
int rotate(int *A, int n, int k)
{
    if (!A) return -1;
    k %= n;
    if (!k) return 0;

    reverse(A, 0, n - 1);
    reverse(A, 0, k - 1);
    reverse(A, k, n - 1);
    return 0;
}