[코딩테스트 입문] Swift - Day 7

1. 2차원으로 만들기
Q. 정수 배열 num_list와 정수 n이 매개변수로 주어집니다. num_list를 다음 설명과 같이 2차원 배열로 바꿔 return하도록 solution 함수를 완성해주세요.

num_list가 [1, 2, 3, 4, 5, 6, 7, 8] 로 길이가 8이고 n이 2이므로 num_list를 2 * 4 배열로 다음과 같이 변경합니다. 2차원으로 바꿀 때에는 num_list의 원소들을 앞에서부터 n개씩 나눠 2차원 배열로 변경합니다.

func solution(_ numList:[Int], _ n:Int) -> [[Int]] {
    (0...(numList.count/n)-1).map { Array(numList[(n * $0)...((n - 1)+(n * $0))]) }
}

2. 공던지기
Q. 머쓱이는 친구들과 동그랗게 서서 공 던지기 게임을 하고 있습니다. 공은 1번부터 던지며 오른쪽으로 한 명을 건너뛰고 그다음 사람에게만 던질 수 있습니다. 친구들의 번호가 들어있는 정수 배열 numbers와 정수 K가 주어질 때, k번째로 공을 던지는 사람의 번호는 무엇인지 return 하도록 solution 함수를 완성해보세요.

func solution(_ numbers:[Int], _ k:Int) -> Int {
    var cnt = numbers.count
    var idx = (k-1)*2
    
    if cnt > idx {
        return numbers[idx]
    } else {
        return numbers[idx%cnt]
    }
}

3. 배열 회전시키기
Q. 정수가 담긴 배열 numbers와 문자열 direction가 매개변수로 주어집니다. 배열 numbers의 원소를 direction방향으로 한 칸씩 회전시킨 배열을 return하도록 solution 함수를 완성해주세요.

func solution(_ numbers:[Int], _ direction:String) -> [Int] {
    var nums = numbers
    direction == "right" ? nums.insert(nums.removeLast(), at: 0) : nums.append(nums.removeFirst())
    return nums
}

 

 

출처: https://school.programmers.co.kr/learn/challenges