개인 일정으로 5일차~14일차까지의 TIL을 작성 못해, 오늘부터 몰아서 작성중이다.
현재 7일차까지 완성했고, 오늘 15일차도 작성하려고 한다.
오늘 문제는 완전 탐색 알고리즘의 카테고리에 속한 문제이다.
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.
1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...
1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.
answers | return |
---|---|
[1,2,3,4,5] | [1] |
[1,3,2,4,2] | [1,2,3] |
입출력 예 #1
따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.
입출력 예 #2
3명의 수포자가 문제를 찍은 규칙이 나와있다. 이것을 보고 문제의 정답 (answers) 리스트가 주어졌을 때, 가장 많은 문제를 맞힌 사람을 리스트 형태로 return 하는 문제이다. 우선 3명의 수포자의 규칙을 리스트로 나타내보자.
num1 = [1,2,3,4,5]
num2 = [2,1,2,3,2,4,2,5]
num3 = [3,3,1,1,2,2,4,4,5,5]
num1의 길이는 5, num2의 길이는 8, num3의 길이는 10이다. 3명의 수포자의 문제 규칙의 길이는 다 다르다. 이것을 동일하게? 문제의 정답 리스트와 비교를 하려면 수포자의 규칙을 순회해서 확인할 수 있도록 해야한다.
num2을 예시로 본다면,
index0 → 2
index1 → 1
index2 → 2
index3 → 3
index4 → 2
index5 → 4
index6 → 2
index7 → 5
입출력 예 첫번째의 answers는 [1,2,3,4,5]이다. 즉, len(answers) == 5이다. answers의 길이에 맞게 num2의 요소를 순회하려면 다음과 같이 진행하면 된다.
answers의 인덱스/ num2의 인덱스
index0 → 1 / index0 →2
index1 → 2 / index1 →1
index2 → 3 / index2 → 2
index3 → 4 / index3 → 3
index4 → 5 / index4 → 2
/ index5 → 4
/ index6 → 2
/ index7 → 5
만약 answers가 더 길다면? answers의 길이가 10, nums의 길이가 5로 다음과 같이 나타낼 수 있다.
index0/index0
index1/index1
index2/index2
index3/index3
index4/index4
index5/index0 == index(5%len(nums))
index6/index1 == index(6%len(nums))
index7/index2 == index(7%len(nums))
index8/index3 == index(8%len(nums))
index9/index4 == index(9%len(nums))
즉, answers의 현재 index에서 수포자의 길이로 나눈 나머지 값의 인덱스에 해당하는 nums를 answers와 비교하면 된다. 이것을 코드로 나타내보자.
def solution(answers): # len(answers) = 5
num1 = [1,2,3,4,5] # len(num1) = 5
num2 = [2,1,2,3,2,4,2,5] # len(num2) = 8
num3 = [3,3,1,1,2,2,4,4,5,5] # len(num3) = 10
score = [0,0,0] # 세 명의 수포자의 점수를 기록하는 리스트
res = [] # 세 명의 수포자 중 가장 많은 문제를 맞춘 수포자를 기록하는 리스트
for i, ans in enumerate(answers): # i = {0,1,2,3,4}, 완전 탐색
# answers의 요소 인덱스를 알기 위해 enumerate()를 사용함
if ans == num1[i%len(num1)]:
score[0] += 1
if ans == num2[i%len(num2)]:
score[1] += 1
if ans == num3[i%len(num3)]:
score[2] += 1
for i,s in enumerate(score):
if s == max(score):
res.append(i+1) # i+1인 이유는 score에서 첫번째 수포자는 score에서 인덱스 값이 0이므로
return res
오늘 푼 문제는 완전 탐색 문제로, 비교적 간단하게 해결할 수 있는 문제였다.
저번주 내내 코테를 제대로 진행하지 못했지만 다시 나아간다!!!