[문제 설명]
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.
[입출력 예]
[입출력 예 설명]
입출력 예 #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]를 반환합니다.
[나의 풀이]
def solution(name, yearning, photo):
result = []
for people in photo:
score = 0
for i in people:
if i in name:
score += yearning[name.index(i)]
result.append(score)
return result
[설명]
1. 사진(photo) 속 인물들 꺼내기
name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
photo = [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]
for people in photo:
print(people)Out[1] :
['may', 'kein', 'kain', 'radi']
['may', 'kein', 'brin', 'deny']
['kon', 'kain', 'may', 'coni']
2. 한 사람씩 꺼내기
name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
photo = [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]
for people in photo:
print()
for i in people:
print(i)Out[2] :
may
kein
kain
radi
may
kein
brin
deny
kon
kain
may
coni
3. score 계산하기
name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
photo = [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]
for people in photo:
print()
score = 0
for i in people:
if i in name:
score += yearning[name.index(i)]
print('name.inedx(i) : ',name.index(i),
' / yearning[name.index(i)] : ', yearning[name.index(i)],
' / score : ',score)Out[3] :
name.inedx(i) : 0 / yearning[name.index(i)] : 5 / score : 5
name.inedx(i) : 1 / yearning[name.index(i)] : 10 / score : 15
name.inedx(i) : 2 / yearning[name.index(i)] : 1 / score : 16
name.inedx(i) : 3 / yearning[name.index(i)] : 3 / score : 19
name.inedx(i) : 0 / yearning[name.index(i)] : 5 / score : 5
name.inedx(i) : 1 / yearning[name.index(i)] : 10 / score : 15
name.inedx(i) : 2 / yearning[name.index(i)] : 1 / score : 1
name.inedx(i) : 0 / yearning[name.index(i)] : 5 / score : 6
4. result를 배열에 담기
name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
photo = [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]
result = []
for people in photo:
score = 0
for i in people:
if i in name:
score += yearning[name.index(i)]
result.append(score)
print(result)Out[4] :
[19, 15, 6]
[다른 사람 풀이]
def solution(name, yearning, photo):
return [sum(yearning[name.index(j)] for j in i if j in name) for i in photo][설명]
name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
photo = [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]
[sum(yearning[name.index(j)] for j in i if j in name) for i in photo]Out[5] :
[19, 15, 6]