[Algorithm] 추억 점수 (w/ Swift)

 

추억 점수

프로그래머스

 

 

1. [name: yearning] Dictionary 생성

    (string 타입 key 값은 그리운 사람의 이름, Int 타입 value 값은 그리움 점수 ) 

2. 사진마다 dict에 대입

3. 합산한 점수들의 배열(yeon)을 출력

 

 

func solution(_ name: [String], _ yearning: [Int], _ photo: [[String]]) -> [Int] {
    var dict = [String: Int]() // 이름과 그리움 값을 저장하는 딕셔너리
    for i in 0..<name.count { // 이름 배열을 순회하며 딕셔너리에 이름과 그리움 값 저장
        dict[name[i]] = yearning[i]
    }
    
    var yeon = [Int]() // 각 사진의 총 그리움 점수를 저장할 배열
    
    for i in 0..<photo.count { // 사진 배열을 순회
        let arr = photo[i] // 현재 사진
        
        // 현재 사진에 있는 각 이름에 대해 그리움 값을 찾아 합산
        let points = arr.map { (name: String) -> Int in
            if let value = dict[name] { // 이름이 딕셔너리에 있으면 해당 그리움 값을 반환
                return value
            } else { // 이름이 딕셔너리에 없으면 0 반환
                return 0
            }
        }
        yeon.append(points.reduce(0, +)) // 현재 사진의 총 그리움 점수를 yeon 배열에 추가
    }
    return yeon // 각 사진의 총 그리움 점수 배열 반환
}

 

---

 

if dict[name] == nil { return 0 } else { return dict[name]! }

 

△ ?? 연산자를 사용하지 않고 조건문을 통해 구현했다. Nil-Coalescing 연산자를 사용하면  부분을 더 간결하게 아래처럼 표현할 수 있다.

return dict[name] ?? 0