배열의 길이에 따라 다른 연산하기

문제

https://school.programmers.co.kr/learn/courses/30/lessons/181854

프로그래머스

 

내 답안 / 다른 사람들의 답안

import Foundation

func solution(_ arr:[Int], _ n:Int) -> [Int] {
    return arr.count % 2 == 0 ? arr.enumerated().map { $0.offset % 2 != 0 ? $0.element + n : $0.element }
    : arr.enumerated().map { $0.offset % 2 == 0 ? $0.element + n : $0.element }
}

- 삼항연산자를 사용해서, count가 홀수인지 짝수인지를 기준으로 값을 반환

- arr.count가 짝수라면 enumerated한 짝에서 offset이 홀수인 경우에 element에 n을 더한 배열을 반환

- arr.count가 홀수라면 enumerated한 짝에서 offset이 짝수인 경우에 element에 n을 더한 배열을 반환

 

// 풀이 1

import Foundation

func solution(_ arr:[Int], _ n:Int) -> [Int] {
    arr.count % 2 == 1 ? arr.enumerated().map{ ($0.offset % 2) == 0 ? $0.element+n : $0.element } : arr.enumerated().map{ ($0.offset % 2) == 1 ? $0.element+n : $0.element }
}

// 풀이 2

import Foundation

func solution(_ arr:[Int], _ n:Int) -> [Int] {
    var arr = arr

    if arr.count % 2 == 0 {
        for i in 0..<arr.count {
            if i % 2 == 1 {
                arr[i] += n
            }
        }
    } else {
        for i in 0..<arr.count {
            if i % 2 == 0 {
                arr[i] += n
            }
        }
    }
    return arr
}

 

 

점수: +1