프로그래머스(Lv1) – 추억점수 문제(파이썬)

by
Programmers
Meet the best programming lessons in videos and practices. Programmers provide programing lessons, algorithm problems, programming competitions, and blockchain materials.

프로그래머스 문제풀이

문제

코딩테스트 연습 – 추억 점수
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 [“may”, “kein”, “kain”]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 [“kali”, “mari”, “don”, “tony”]이고 [“kali”, “mari”, “don”]의 그리움 점수가 각각 [11점, 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]의 원소들은 중복된 값이 들어가지 않습니다.

입출력 예

nameyearningphotoresult
[“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]

입출력 예 설명

입출력 예 #1

첫 번째 사진 속 “may”, “kein”, “kain”, “radi”의 그리움 점수를 합치면 19(5 + 10 + 1 + 3)점 입니다. 두 번째 사진 속 그리워하는 사람들인 “may”와 “kein”의 그리움 점수를 합치면 15(5 + 10)점입니다. 세 번째 사진의 경우 “kain”과 “may”만 그리워하므로 둘의 그리움 점수를 합한 6(1 + 5)점이 사진의 추억 점수입니다. 따라서 [19, 15, 6]을 반환합니다.

입출력 예 #2

첫 번째 사진 속 그리워하는 사람들인 “kali”, “mari”, “don”의 그리움 점수를 합치면 67(11 + 1 + 55)점입니다. 두 번째 사진 속엔 그리워하는 인물이 없으므로 0점입니다. 세 번째 사진 속 그리워하는 사람은 “don”만 있으므로 55점입니다. 따라서 [67, 0, 55]를 반환합니다.

입출력 예 #3

설명 생략

문제 개념해석

이번 문제를 요약하자면,

  1. 사람별로 각 점수가 존재한다.(리스트에 없는 사람은 0점)
  2. 사람이름에 맞는 점수를 부여한다.
  3. 합산된 점수를 구한다.

문제 풀이전에 지문과 제한사항을 반드시 읽으시길 바랍니다.

나의 풀이

이번문제는 name과 yearning 리스트를 dict로 만들어주고, photo라는 이중리스트의 각 리스트요소를 점수로 환산하여 SUM함수를 사용하여 합산 결과를 출력해주었다.

def solution(name, yearning, photo):
    answer = []
    dic = dict(zip(name, yearning))                            # name과 yearning을 zip으로 묶어서 dict화 시켜줌
    for i in photo:                                            # photo의 요소반복
        answer.append(sum([dic[j] for j in i if j in dic]))    # 리스트요소의 요소값이 dic에 존재하는지 확인  
                                                               # 점수로 환산하여 합산
                                                               # 리스트에 추가
    return answer

다른 사람의 풀이

다른 풀이 중 한줄짜리 코드를 가져와보았습니다. 아래 풀이는 인덱스를 이용하여 풀었네요. 

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

관련 링크

문제해결과정에서 기초가 부족하다고 느껴지시면 아래 정리 글을 참고해주세요.
추가적으로 궁금한 사항이나 글에 대한 개선점 알려주시면 피드백 적극 반영하도록 하겠습니다.

Python 문법 총 정리1 – Zzrany Blog
여기는 Python을 공부하며 학습한 내용을 정리하는 블로그입니다. 해당 포스팅에서는 개발시 참고하실 수 있도록 Python 문법들을 한눈에 보이도록 필요한 내용만 간략하게 정리해보았습니다.
Python 문법 총 정리2 – Zzrany Blog
여기는 Python을 공부하며 학습한 내용을 정리하는 블로그입니다. 해당 포스팅에서는 개발시 참고하실 수 있도록 Python 문법들을 한눈에 보이도록 필요한 내용만 간략하게 정리해보았습니다.