数组循环右移

设计一个算法，把一个含有N个元素的数组循环右移K位，要求时间复杂度为O（N），且只容许使用两个附加变量。

不合题意的解法以下：

咱们先试验简单的办法，能够每次将数组中的元素右移一位，循环K次。abcd1234→4abcd123→34abcd12→234abcd1→1234abcd。伪代码以下：

代码清单2-33

 

 

RightShift(int* arr, int N, int K)

{

    while(K--)

    {

        int t = arr[N - 1];

        for(int i = N - 1; i > 0; i --)

            arr[i] = arr[i - 1];

        arr[0] = t;

    }

}

虽然这个算法能够实现数组的循环右移，可是算法复杂度为O（K * N），不符合题目的要求，须要继续往下探索。

 

分析与解法

假如数组为abcd1234，循环右移4位的话，咱们但愿到达的状态是1234abcd。不妨设K是一个非负的整数，当K为负整数的时候，右移K位，至关于左移（-K）位。左移和右移在本质上是同样的。

【解法一】

你们开始可能会有这样的潜在假设，K<N。事实上，不少时候也的确是这样的。但严格地说，咱们不能用这样的“惯性思惟”来思考问题。尤为在编程的时候，全面地考虑问题是很重要的，K多是一个远大于N的整数，在这个时候，上面的解法是须要改进的。

仔细观察循环右移的特色，不难发现：每一个元素右移N位后都会回到本身的位置上。所以，若是K > N，右移K-N以后的数组序列跟右移K位的结果是同样的。进而可得出一条通用的规律：右移K位以后的情形，跟右移K’= K % N位以后的情形同样。

代码清单2-34

RightShift(int* arr, int N, int K)

{

    K %= N;

    while(K--)

    {

        int t = arr[N - 1];

        for(int i = N - 1; i > 0; i --)

            arr[i] = arr[i - 1];

        arr[0] = t;

    }

}

可见，增长考虑循环右移的特色以后，算法复杂度降为O（N^２），这跟K无关，与题目的要求又接近了一步。但时间复杂度还不够低，接下来让咱们继续挖掘循环右移先后，数组之间的关联。

【解法二】

假设原数组序列为abcd1234，要求变换成的数组序列为1234abcd，即循环右移了4位。比较以后，不难看出，其中有两段的顺序是不变的：1234和abcd，可把这两段当作两个总体。右移K位的过程就是把数组的两部分交换一下。变换的过程经过如下步骤完成：

1.   逆序排列abcd：abcd1234 → dcba1234；

2.   逆序排列1234：dcba1234 → dcba4321；

3.   所有逆序：dcba4321 → 1234abcd。

伪代码能够参考以下：

代码清单2-35

Reverse(int* arr, int b, int e)

{

    for(; b < e; b++, e--)

    {

        int temp = arr[e];

        arr[e] = arr[b];

        arr[b] = temp;

    }

}

RightShift(int* arr, int N, int k)

{

    K %= N;

    Reverse(arr, 0, N – K - 1);

    Reverse(arr, N - K, N - 1);

    Reverse(arr, 0, N - 1);

}

这样，咱们就能够在线性时间内实现右移操做了。