数组循环移位

2020-08-23  本文已影响0人  LEO_青蛙

设计一个算法,把一个含有N个元素的数组循环右移K位,要求时间复杂度为O(N),且只允许使用两个附加变量。
1、简单的办法,每次将数组中的元素右移一位,循环K次。

void RightShift(int *arr, int N, int K)
{
  K %= N;
  while(K--)
  {
    int temp = arr[N-1];
    for(int i=N-1; i>0; --i)
    {
      arr[i] = arr[i-1];
    }
    arr[0] = temp;
  }
}

算法复杂度为O(K*N)
2、优化的方法,将数组分成两个整体,右移K位的过程就是把数组的两部分交换位置。

void Reverse(int *arr, int b, int e)
{
  for(; b<e; b++, e--)
  {
    int temp = arr[e];
    arr[e] = arr[b];
    arr[b] = temp;
  }
}

void RightShift(int *arr, int N, int K)
{
  K %= N;
  Reverse(arr, 0, N-K-1);
  Reverse(arr, N-K-1, N-1);
  Reverse(arr, 0, N-1);
}

算法复杂度为O(N);

上一篇 下一篇

猜你喜欢

热点阅读