Level1. 모의고사

[jeeyeonLIM]

문제 설명

• 수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 함수를 작성해주세요.

제한 조건

• 시험은 최대 10,000 문제로 구성되어있습니다.
• 문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

입출력 예

answers	return
[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

입출력 예 설명

입출력 예 #1

수포자 1은 모든 문제를 맞혔습니다. 수포자 2는 모든 문제를 틀렸습니다. 수포자 3은 모든 문제를 틀렸습니다. 따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.

입출력 예 #2

모든 사람이 2문제씩을 맞췄습니다.

[jeeyeonLIM]

나의코드

def solution(answers):
    answer = [0,0,0]
    s1=[1,2,3,4,5]
    s2=[2,1,2,3,2,4,2,5]
    s3=[3,3,1,1,2,2,4,4,5,5]

    for i in range(len(answers)):
        if answers[i] == s1[int(i%len(s1))]:answer[0]+=1
        if answers[i] == s2[int(i%len(s2))]:answer[1]+=1
        if answers[i] == s3[int(i%len(s3))]:answer[2]+=1

    m=max(answer)
    return [i+1 for i, j in enumerate(answer) if j == m]

• 고민이였던 점은 수포자들이 찍는 패턴을 입력하는 데 있어서 반복되는 구간만을 입력했는데, 이부분을 어떤 방법으로 바꿀 수 있을까..였다.
• 또 if문 중복했는데 다른 방법은 없을까하는 의문이다.
• 많이 연습해야 겠다. 부족함을 또 한번 느껴본다.

[jeeyeonLIM]

다른사람들 코드

• 모범답안 코드가 나랑 비슷했다. 패턴은 직접 써줘야 하는 게 맞다. 단 answer값과 수포자들의 답을 확인할때도 enumerate를 사용해줬다.
• 아래는 가장 간결한 코드.

  def solution(answers):
  p = [[1, 2, 3, 4, 5],
       [2, 1, 2, 3, 2, 4, 2, 5],
       [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]]
  s = [0] * len(p) # 자동 확장됨.
  for q, a in enumerate(answers):
      for i, v in enumerate(p):
          #print(a, v[q % len(v)])
          if a == v[q % len(v)]:
              s[i] += 1
  return [i + 1 for i, v in enumerate(s) if v == max(s)]

• answer 값을 수포자 1,2,3 값과 비교해줘야 하므로 가장 겉 for문을 answer로 줬고, 값을 하나하나 뜯어서 수포자들것과 비교해줬다.
• 쉽게 풀어서 설명하자면, q는 answer 값의 index, a는 answers 값자체, i는 몇번째 수포자인지를 나타내주는 index, v는 수포자들 값이 들어있는 list 전체이다. 또한, a==v[q%len(v)] 를 봤을 때, answer 값과 n번째 수포자의 index 값을 처음부터 하나하나 비교해준다.
• Ex. answer = [1,2,3,4,5] 값이 들어왔다고 했을 때, 각각 a, v[q%len(v)] 값을 print 해봤을 때는 다음과 같다.

  1 1 # answer[0] 값과 p[0][0] 비교
  1 2 # answer[0] 값과 p[1][0] 비교
  1 3 # answer[0] 값과 p[2][0] 비교
  2 2
  2 1
  2 3
  3 3
  3 2
  3 1
  4 4
  4 3
  4 1
  5 5 # answer[4] 값과 p[0][4] 비교
  5 2 # answer[4] 값과 p[1][4] 비교
  5 2 # answer[4] 값과 p[2][4] 비교