출처 : 프로그래머스
문제는 여기에
문제
- 그리워하는 사람의 이름이 담긴 배열 name 이 주어진다
- 각 사람별 그리움 점수가 담긴 배열 yearning 이 주어진다.
- 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo 가 주어진다.
- 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 된다.
- 사진들의 추억 점수를 photo 에 주어진 순서대로 배열에 담아 return 하는 solution 함수를 완성해라
제한사항
입출력 예
접근
- name에 나온 사람의 추억점수가 yearning에 순서대로 입력되어 있으니 for문과 딕셔너리를 이용해서 이름:점수 로 만들어야겠다.
- 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함수는 추후에 개념 정리를 해야한다고 느꼈다.
'🧑💻 Coding Test > 프로그래머스' 카테고리의 다른 글
[프로그래머스 Level 1] 과일 장수 (Python) (0) | 2023.08.31 |
---|---|
[프로그래머스 Level1] 삼총사 (Python) (0) | 2023.08.30 |
[프로그래머스 Level 1] 달리기 경주 (Python) (0) | 2023.08.24 |
[프로그래머스 Level 1] 크기가 작은 부분 문자열 (Python) (0) | 2023.08.22 |
[프로그래머스 Level 1] 나머지가 1이 되는 수 찾기 (Python) (0) | 2023.08.22 |