Python程序在向右旋转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])。除此以外,
下面是上述方法的实现。
Python3
# Python3 implementation of right rotation
# of an array K number of times
# Function to rightRotate array
def RightRotate(a, n, k):
# If rotation is greater
# than size of array
k = k % n;
for i in range(0, n):
if(i < k):
# Printing rightmost
# kth elements
print(a[n + i - k], end = " ");
else:
# Prints array after
# 'k' elements
print(a[i - k], end = " ");
print("
");
# Driver code
Array = [ 1, 2, 3, 4, 5 ];
N = len(Array);
K = 2;
RightRotate(Array, N, K);
# This code is contributed by Code_Mech
输出:
4 5 1 2 3
时间复杂度:O(n)
辅助空间:O(1)
更多细节请参考完整文章打印阵列右转K次!