[프로그래머스] Swift - 추억점수 (LV.1)

👆문제풀러 가기👆

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["may", "kein", "kain"]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 ["kali", "mari", "don", "tony"]이고 ["kali", "mari", "don"]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, "tony"는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점입니다.

그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해주세요.

제한 사항

• 3 ≤ name의 길이 = yearning의 길이≤ 100
  • 3 ≤ name의 원소의 길이 ≤ 7
  • name의 원소들은 알파벳 소문자로만 이루어져 있습니다.
  • name에는 중복된 값이 들어가지 않습니다.
  • 1 ≤ yearning[i] ≤ 100
  • yearning[i]는 i번째 사람의 그리움 점수입니다.
• 3 ≤ photo의 길이 ≤ 100
  • 1 ≤ photo[i]의 길이 ≤ 100
  • 3 ≤ photo[i]의 원소(문자열)의 길이 ≤ 7
  • photo[i]의 원소들은 알파벳 소문자로만 이루어져 있습니다.
  • photo[i]의 원소들은 중복된 값이 들어가지 않습니다.

---

풀이 [ 메모리: 17mb, 최대시간: 2.17ms ]

func solution(_ name:[String], _ yearning:[Int], _ photo:[[String]]) -> [Int] {
    var missed = [String: Int]()

    // 그리운 사람(key)별 그리움 점수(value) 할당
    for (person, point) in zip(name, yearning) {
        missed[person] = point
    }

    var result = [Int]()

    // 사진 속 사람들
    for people in photo {
        var sum = 0

        for person in people {
            // 그리운 사람인지 확인 후, 그리움 점수 합산
            if let yearningPoint = missed[person] {
                sum += yearningPoint
            }
        }
        result.append(sum)
    }

    return result
}

이 문제는 문제가 스스로 어떻게 풀어야 하는지 설명해주는듯한 느낌을 받았다.

문제 난이도가 낮았기에 코드가 최대한 쉽게 읽힐 수 있도록 변수에 신경을 써서 작성했다.

 

1. 그리운 사람과 그리움 점수는 함께 가야하기 때문에 Dictionary의 키, 값으로 만들었다.

2. 사진 속 사람들 중 점수를 할당하지 않은 사람들이 일부 섞여 있다. (= 그립지 않은 사람들 = 그리움 점수 0 = 계산할 필요 없음)

3. 사진 속 사람들 중 그리운 사람들만 선택적으로 그리움 점수를 합산한다.

 

개인적인 생각이지만 한 함수가 10줄이 넘어가면 조금,, 부담스럽다. 그래서 코드를 나눠봤다.

이 문제에서는 이 이상 나누면 조금 과하게 나눈 것 같았고, 이정도가 적당하지 않을까 싶다.

하지만 실제 앱 구현을 한다면 solution에는 별다른 구현이 남지 않고, 변수명만을 보고 읽을 수 있는 코드가 좋은 코드가 아닐까 싶다.

func solution(_ name:[String], _ yearning:[Int], _ photo:[[String]]) -> [Int] {
    var missed = [String: Int]()

    for (person, point) in zip(name, yearning) {
        missed[person] = point
    }

    var result = [Int]()

    for people in photo {
        result.append(sumOfYearningPoint(people, missed))
    }

    return result
}

func sumOfYearningPoint(_ people: [String], _ missed: [String: Int]) -> Int{
    var sum = 0

    for person in people {
        if let yearningPoint = missed[person] {
            sum += yearningPoint
        }
    }
    return sum
}