Lv1. 추억점수

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.

입출력 예

name	yearning	photo	result
["may", "kein", "kain", "radi"]	[5, 10, 1, 3]	[["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]	[19, 15, 6]
["kali", "mari", "don"]	[11, 1, 55]	[["kali", "mari", "don"], ["pony", "tom", "teddy"], ["con", "mona", "don"]]	[67, 0, 55]
["may", "kein", "kain", "radi"]	[5, 10, 1, 3]	[["may"],["kein", "deny", "may"], ["kon", "coni"]]	[5, 15, 0]

---

내 풀이

def solution(name, yearning, photo):
    answer = []
    # assign nums to dict
    dict = {}
    for i in range(len(name)):
        dict[name[i]] = yearning[i]
    # photos
    for lst in photo:
        score = 0
        for person in lst: 
            if person in list(dict.keys()):
                score += dict[person]
        answer.append(score)
    return answer

정석에 가까운 풀이지 않나.. 뭐 그런 생각.. 인덱스를 찾아서 구할 까 생각도 했는데 그냥 dictionary 만들어서 참조하는게 제일 쉬울 것 같아서 선택했다 간단한 문제라 크게 어렵진 않았는데 다른 사ㅏㄹㅁ 들 풀이를 보다보니 적어두고 싶어서 남겨둠

다른 사람 풀이

def solution(이름, 점수, 사진):
    return [sum(점수[이름.index(j)] for j in i if j in 이름) for i in 사진]

한줄로 짜서 대단하다.. 싶었던 풀이 내가 쓰고 싶었던 것도 이런 코드 이긴 했는데 인덱스 사용해서 했어도 이렇게까지는 진행 안됐을듯 ㅎㅎ..

def solution(name, yearning, photo):
    dictionary = dict(zip(name,yearning))
    scores = []
    for pt in photo:
        score = 0
        for p in pt:
            if p in dictionary:
                score += dictionary[p]
        scores.append(score)
    return scores

zip으로 바로 처리한게 신기해서 적어둔다 zip 하면 오브젝트 타입이 달라지니까 일부러 안쓴건데 그냥 리스트로 overwrite 하면 되는거였음..