-
Notifications
You must be signed in to change notification settings - Fork 73
/
Copy pathcyclicrotation.py
40 lines (31 loc) · 1.13 KB
/
cyclicrotation.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
A zero-indexed array A consisting of N integers is given.
Rotation of the array means that each element is shifted right by one index,
and the last element of the array is also moved to the first place.
For example, the rotation of array A = [3, 8, 9, 7, 6] is [6, 3, 8, 9, 7].
The goal is to rotate array A K times; that is, each element of A will be shifted
to the right by K indexes.
Write a function:
def solution(A, K):
that, given a zero-indexed array A consisting of N integers and an integer K,
returns the array A rotated K times.
For example, given array A = [3, 8, 9, 7, 6] and K = 3, the function should
return [9, 7, 6, 3, 8].
Assume that:
N and K are integers within the range [0..100];
each element of array A is an integer within the range [−1,000..1,000].
In your solution, focus on correctness. The performance of your solution will
not be the focus of the assessment.
"""
# 75%
# def solution(A, K):
# # write your code in Python 2.7
# n = K - 1
# return A[n:] + A[:n]
# 100%
def solution(A, K):
n = len(A)
B = [None] * n
for i in range(0, n):
B[(i + K) % n] = A[i]
return B