프로그래머스 코딩테스트 연습문제 : 추억 점수

연구실 분들이랑 일주일에 한 번씩 "프로그래머스"라는 사이트에서 문제를 푸는 코딩 스터디를 하기로 했다. 쉬운 난이도로 시작해서, 하는 거 보고 난이도를 올려갈 생각이라고 한다. 이 블로그에서는 [코딩연습] 섹션에 기록하도록 하겠다.

 

첫 주차라서, 이번 주 문제는 쉬웠다. 쉬울수록 천하제일 숏코딩대회가 되는 코딩판

 

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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 함수를 완성해주세요.

 

 

크게 어려운 점은 없으며 기본적인 자료형의 값과 인덱스에 잘 접근할 수 있는지 확인하는 문제다.

 

나의 풀이

def solution(name, yearning, photo):

    dic = dict(zip(name, yearning))
    answer = list(map(sum, [[(dic[person] if person in name else 0) for person in pic] for pic in photo]))
    
    return answer

 

 

다른 사람 풀이

 

어차피 yearning 순서가 name 순서와 일치하니까 바로 index를 이용해 한 줄만에 푸셨다. 뭐 짧긴 하군요...

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