C++程序在右旋转K次后打印数组
给定一个大小为 N 的数组和一个值K ,我们需要围绕它右旋数组。如何快速打印正确的旋转数组?
例子 :
Input: Array[] = {1, 3, 5, 7, 9}, K = 2.
Output: 7 9 1 3 5
Explanation:
After 1st rotation - {9, 1, 3, 5, 7}
After 2nd rotation - {7, 9, 1, 3, 5}
Input: Array[] = {1, 2, 3, 4, 5}, K = 4.
Output: 2 3 4 5 1
方法:
- 我们将首先取 K 乘以 N (K = K % N) 的 mod,因为在每 N 次旋转后,数组将变得与初始数组相同。
- 现在,我们将数组从 i = 0 迭代到 i = N-1 并检查,
- 如果i < K ,打印最右边的第 K 个元素 (a[N + i -K])。除此以外,
- 在 'K' 个元素 (a[i – K]) 之后打印数组。
- 如果i < K ,打印最右边的第 K 个元素 (a[N + i -K])。除此以外,
下面是上述方法的实现。
C++
// C++ implementation of right rotation
// of an array K number of times
#include
using namespace std;
// Function to rightRotate array
void RightRotate(int a[], int n, int k)
{
// If rotation is greater
// than size of array
k = k % n;
for(int i = 0; i < n; i++)
{
if(i < k)
{
// Printing rightmost
// kth elements
cout << a[n + i - k] << " ";
}
else
{
// Prints array after
// 'k' elements
cout << (a[i - k]) << " ";
}
}
cout << "
";
}
// Driver code
int main()
{
int Array[] = { 1, 2, 3, 4, 5 };
int N = sizeof(Array) / sizeof(Array[0]);
int K = 2;
RightRotate(Array, N, K);
}
// This code is contributed by Surendra_Gangwar
输出:
4 5 1 2 3
时间复杂度:O(n)
辅助空间:O(1)
更多细节请参考完整文章打印阵列右转K次!