[프로그래머스 Level 1] 추억점수 (Python)

출처 : 프로그래머스

 

문제는 여기에

---

문제

• 그리워하는 사람의 이름이 담긴 배열 name 이 주어진다
• 각 사람별 그리움 점수가 담긴 배열 yearning 이 주어진다.
• 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo 가 주어진다.
• 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 된다.
• 사진들의 추억 점수를 photo 에 주어진 순서대로 배열에 담아 return 하는 solution 함수를 완성해라

제한사항

입출력 예

접근

1. name에 나온 사람의 추억점수가 yearning에 순서대로 입력되어 있으니 for문과 딕셔너리를 이용해서 이름:점수 로 만들어야겠다.
2. for문을 돌며 photo에 있는 사람들이 딕셔너리의 키의 값에 존재한다면, 그 키에 해당하는 값을 answer에 추가해야겠다.

코드

def solution(name, yearning, photo):
    dic = {}
    answer = [0 for i in range(len(photo))]
    
    for i in range(len(name)):
        dic[name[i]] = yearning[i]
        
    for i in range(len(photo)):
        for j in range(len(photo[i])):
            if photo[i][j] in dic.keys():
                answer[i] += dic[photo[i][j]]
            else:
                continue
    return answer

고찰

우선 불필요한 코드가 있는 것 같다.

1. 리스트를 0으로 초기화를 해야 할 필요가 있을까?    <= 초기 코드 작성할 때, append를 사용해보니 인덱스 오류가 나서 바꾼 것이다.

2. 이중 배열을 for문 돌때, photo[i][j]와 같이 하지 않아도 된다.

    첫 for문에서 리스트 형식으로 반복하고, 두 번째 for문에서 리스트 내의 원소를 반복하면 된다.

3. 접근 (1)에서 사람의 이름과 점수가 순서대로 입력되어 있으니 사전을 만들지 않아도 된다. 인덱스가 같다는 것으로 접근하면 더 쉽다.

 

수정한 코드

def solution(name,yearning,photo):
    answer = []
    
    for i in photo:
        score = 0
        for j in range(len(name)):
            if name[j] in i:
                score += yearning[j]
        answer.append(score)
    return answer

이렇게 이중 배열인 photo 자체를 반복시키면 i에는 (단일?) 배열이 들어간다. if 문에서도 보면 사진에 있는 사람들이 그리운 사람 명단에 있는지를 보는 것이 아니라 그 반대인 그리운 사람 명단이 그 사진에 들어가 있는지로 기준을 바꾸면 더 쉬워지는 것을 알게 되었다.

배열의 인덱스 순서가 같은 관계가 있어도 딕셔너리를 생각하기보단 같은 인덱스로 접근한다고 생각해보자.

마지막으로 엄청난 사람의 코드 2개를 보고 마무리 하겠다. . .

 

다른 사람의 코드

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

def solution(name, yearning, photo):
    answer = []
    for x in photo:
        answer.append(sum([ yearning[name.index(y)] for y in x if y in name]))
    return answer

def solution(name, yearning, photo):
    score_dict = dict(zip(name,yearning))
    return [sum(map(lambda x: score_dict[x] if x in score_dict else 0,p)) for p in photo]

이렇게 짧게 쓰는 사람들은 평소에 뭐 먹고 지내는지 궁금하다.

dict함수와 zip함수는 추후에 개념 정리를 해야한다고 느꼈다.