[Python] 프로그래머스, 추억 점수 Lv.1 (feat. zip, dictionary, 한 줄 코드, 파이썬)

반짝반짝 알고리즘, 인프준

 

 

 

 

 

 

 

 

 

 

추억 점수

문제 링크

프로그래머스

 

 

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매기려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다.

 

예를 들어 사진 속 인물의 이름이 ["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 함수를 완성해 주세요.

 

 

제한 조건

1) 3 <= name의 길이 = yearning의 길이 <= 100

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

 

2) 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 ]

 

입출력 예 설명

입출력 예#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]를 반환합니다.

 

 

 

 

 

 

---

 

 

 

 

 

 

해결 과정

1. 결괏값 선언, 이름과 그리움 점수는 길이가 같다는 조건이 있으므로 zip()을 활용하여 묶어주어 info라는 딕셔너리를 만들어줍니다.

 

2. 사진들을 확인 할 for loop을 돌려줍니다.

 

3. 각 사진별로 사람들의 그리움 점수를 만들어 줄 score 변수를 선언해줍니다.

 

4. 사진 속 사람들 각각 점수를 누적해주기 위해 for loop을 돌려줍니다.

 

5. 사람들의 그리움 점수를 score에 누적해주고 get method의 기능을 활용하여 없는 사람은 0으로 세팅해줍니다.

 

[Python] Dictionary get method (파이썬 딕셔너리 팁)

 

6. 결괏값 배열에 담아주고 반환하여 해결합니다.

 

 

 

 

제출 코드

def solution(name, yearning, photo):
    # 1. 반환할 결괏값 선언과 이름과 그리움 점수를 묶어준다.
    result = []
    info = dict(zip(name, yearning))
    
    # 2. 사진 속 사람들을 확인할 for loop
    for people in photo:
        
        # 3. 그리움 점수 누적을 위한 변수 선언
        score = 0
        
        # 4. 사람들 각각의 점수를 확인할 for loop
        for person in people:
            # 5. 사람 별 점수를 누적, 없는 사람은 0
            score += info.get(person, 0)
        # 6. 결과 배열에 담아주기
        result.append(score)
    
    return result

 

 

 

 

 

 

 

 

---

 

 

 

 

 

 

 

또 다른 풀이

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

 

가장 인상깊었던 sum()과 index()를 활용하여 한 줄로 구현한 코드입니다.

 

 

 

 

 

 

 

 

 

 

---

 

 

 

 

 

 

 

 

 

 

출처: 프로그래머스 코딩 테스트 연습,https://school.programmers.co.kr/learn/challenges