알고리즘(백준, 프로그래머스)/[프로그래머스] Lv1

[프로그래머스] 정렬 / Lv1. 추억 점수- 파이썬(Python)

마법사 코딩공주 2023. 5. 17. 17:45
728x90
반응형

📝 문제설명

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

첫 번째 사진 속 "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]를 반환합니다.

 


👉 문제풀이

  • 딕셔너리의  key, value 값으로 문제를 처리해야겠다고 생각하여 빈 딕셔너리를 만들어 새로운 딕셔너리 dict1을 만듬
  • 그리고 photo 이중 리스트에서 값을 비교하기 위해 2중 반복문을 사용
  • people에는 ['may', 'kein', 'kain', 'radi'] 'may', 'kein', 'brin', 'deny'] ['kon', 'kain', 'may', 'coni']가 저장되고,
  • person에는 people리스트의 요소가 저장된다. may, kein, kain, radi ...
  • 그리고 if문을 통해 name과 person이름이 같을 경우 result  변수에 그게 해당하는 value값을 더한다.
  • person 반목문이 종료될 때는 모두 더한 값을 answer리스트에 추가한다.
def solution(name, yearning, photo):
    answer = []
    dict1 = {}
    # 딕셔너리 만들기
    for i in range(len(name)):
        dict1[name[i]] = yearning[i]
    
    # 비교하기
    for people in photo:
        result = 0
        for person in people:
            if person in name:
                result += dict1[person]
        answer.append(result)
    return answer

👉 다른사람 문제풀이(zip 함수 활용)

  • 나는 for문을 통해 딕셔너리를 생성했는데 .. zip을 활용하여 간단하게 새로운 딕셔너리를 만들 수 있었다.
def solution(name, yearning, photo):
    dictionary = dict(zip(name,yearning))
    scores = []
    for pt in photo:
        score = 0
        for p in pt:
            if p in dictionary:
                score += dictionary[p]
        scores.append(score)
    return scores
728x90
반응형