数组循环移位(双指针+分治)

题目描述
把一个含有N个元素的数组循环右移K位,要求算法时间复杂度为O(n),且只允许使用两个附加变量。

题解
分析:这道题目最容易想到的思路就是每次将数组中的元素右移一位,循环K次。先把这种思路实现下再接着分析:

void rightShift(int* arr, int N, int k)
{
	 int k = k % N; //循环移动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(kN),不符合题目条件。

其实这里可以借鉴分治的思想,循环移位的本质规律:右移K位的过程就是把数组的两部分交换一下。
例如:12345678,循环右移3位,实际上首先令前5位反转54321,然后令最后三位反转876,此时数组变为54321876,整体再反转一下,就变为了67812345,即循环右移3位。

//此处的反转,其实直接通过双指针,交换左指针与右指针的元素就可以了。
void reverseArr(int* arr, int left, int right)
{
     while(left < right)
     {
          int temp = arr[left];
          arr[left] = arr[right];
          arr[right] = temp;
          left++;
          right--;
     }
}
//分治循环移位
void rightShift2(int* arr, int N, int k)
{
     k = k % N;
     reverseArr(arr, 0, N-1-k);
     reverseArr(arr, N-k, N-1);
     reverseArr(arr, 0, N-1);
}

此时的时间复杂度就变为O(2N)了,此时满足条件。


扩展问题:如果是循环左移呢?其实有了上面的理解,只需要修改下反转的元素就可以了,对前k个元素反转,再反转剩余的元素,最后再反转整个数组。

void leftShift(int* arr, int N, int k)
{
     k = k % N;
     reverseArr(arr, 0, k-1);
     reverseArr(arr, k, N-1);
     reverseArr(arr, 0, N-1);
}

完整的本地测试代码:

#include<stdio.h>
void 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;
     }

}
void reverseArr(int* arr, int left, int right)
{
     while(left < right)
     {
          int temp = arr[left];
          arr[left] = arr[right];
          arr[right] = temp;
          left++;
          right--;
     }
}
void rightShift2(int* arr, int N, int k)
{
     k = k % N;
     reverseArr(arr, 0, N-1-k);
     reverseArr(arr, N-k, N-1);
     reverseArr(arr, 0, N-1);
}
void leftShift(int* arr, int N, int k)
{
     k = k % N;
     reverseArr(arr, 0, k-1);
     reverseArr(arr, k, N-1);
     reverseArr(arr, 0, N-1);
}

int main()
{
     int a[8] = {1, 2, 3, 4, 5, 6, 7, 8};
     int N = 8;
     int k = 3;
     leftShift(a, N, k);
     for(int i=0; i < N; i++)
     {
          printf("%d ", a[i]);
     }
}
  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

爱学习的贝塔

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值