프로그래머스 Lv.1 추억 점수

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]의 원소들은 중복된 값이 들어가지 않습니다.

2. 풀이

1차 시도 (성공)

def solution(name, yearning, photo):
    answer = []
    # name과 yearning을 키-값으로 짝지어서 딕셔너리 생성
    name_score = dict(zip(name, yearning))
    # 점수 저장할 tmp 값 생성
    tmp = 0; 
    for i in range(len(photo)):
        for j in range(len(photo[i])):
            # 만약 없는 이름이면 넘기기
            if photo[i][j] not in name_score:
                continue
            else:
                tmp += name_score[photo[i][j]]
        answer.append(tmp)
        # tmp 값 reset
        tmp = 0
    
    return answer

• 생각보다 쉬운 문제였다. 아마 단순히 리스트에서 찾는 방법을 선택했으면 시간 초과가 났을 것 같다.
  

3. Lv.up

1. 키-값 짝지어서 딕셔너리에 추가하기

words = ["may", "kein", "kain", "radi"]
numbers = [5, 10, 1, 3]

dictionary = dict(zip(words, numbers))

2. 리스트 원소 append
   list.append(3)

4. Ref.

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

• 파이썬은 매개변수에 한글도 쓸 수 있구나..
• 근데 이런식의 코드 작성은 짧은건 좋아도 다른 사람들이 이해하기에 되게 어려운 것 같다..
• sum()이정도는 들고 가도 좋을 듯 싶다. 리스트, 튜플 등을 sum해주는 함수이고, 내용물이 아무것도 없으면 error가 아니라 0으로 반환한다는 특징이 있다.