-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcodility_CyclicRotation_2-2.swift
More file actions
73 lines (67 loc) · 2.24 KB
/
codility_CyclicRotation_2-2.swift
File metadata and controls
73 lines (67 loc) · 2.24 KB
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
//
// codility_CyclicRotation_2-2.swift
// algo
//
// Created by 이재은 on 09/10/2019.
// Copyright © 2019 이재은. All rights reserved.
//
import Foundation
// codility CyclicRotation Lesson 2 - 2
//An 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 moved to the first place. For example, the rotation of array A = [3, 8, 9, 7, 6] is [6, 3, 8, 9, 7] (elements are shifted right by one index and 6 is moved to the first place).
//
//The goal is to rotate array A K times; that is, each element of A will be shifted to the right K times.
//
//Write a function:
//
//class Solution { public int[] solution(int[] A, int K); }
//
//that, given an array A consisting of N integers and an integer K, returns the array A rotated K times.
//
//For example, given
//
// A = [3, 8, 9, 7, 6]
// K = 3
//the function should return [9, 7, 6, 3, 8]. Three rotations were made:
//
// [3, 8, 9, 7, 6] -> [6, 3, 8, 9, 7]
// [6, 3, 8, 9, 7] -> [7, 6, 3, 8, 9]
// [7, 6, 3, 8, 9] -> [9, 7, 6, 3, 8]
//For another example, given
//
// A = [0, 0, 0]
// K = 1
//the function should return [0, 0, 0]
//
//Given
//
// A = [1, 2, 3, 4]
// K = 4
//the function should return [1, 2, 3, 4]
//
//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.
//
//Copyright 2009–2020 by Codility Limited. All Rights Reserved. Unauthorized copying, publication or disclosure prohibited.
// 주어진 배열을 주어진 K만큼 오른쪽으로 회전시킨 배열을 반환
public func solution(_ A : [Int], _ K : Int) -> [Int] {
var A = A
if A.count != 0 {
let k = K % A.count
for _ in 0..<k {
if let last = A.last {
A.insert(last, at: 0)
A.removeLast()
}
}
}
return A
}
print(solution([3, 8, 9, 7, 6], 3)) // [9, 7, 6, 3, 8]
print(solution([0,0,0], 1)) // [0,0,0]
print(solution([1,2,3,4], 4)) // [1,2,3,4]
print(solution([], 0))
print(solution([1], 0))
print(solution([], 4))