프로그래머스_완전탐색_모의고사

문제를 풀면서도 이건 좀 아니다.. 하면서 풀었던 문제 간단해보이는 문제가 60줄이나작 된다. 확실히 코드 자체의 시간 효율성도 떨어지고 작업 효율성.. 그러니까 이렇게 풀면 풀이시간이 엄청 오래걸릴것같다. 오류가 나도 찾기가 힘들듯..

좀 더 깔끔하게 풀 수 있는 방법을 고민해보면 좋을 것 같다.

---

문제설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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문제씩을 맞췄습니다.

---

1. 내 풀이

def solution(answers):
    sol = []
    num_q = len(answers)
    a1,a2,a3 = [], [], []
    cnt1, cnt2, cnt3 = 0, 0, 0
    for i in range(0, num_q):
        a1.append(i%5+1)

        if i%2 ==0:
            a2.append(0)
        else:
            if i%2 ==0 :
                a2.append(2)
            elif i%8 ==1:
                a2.append(1)
            elif i%8 ==3:
                a2.append(3)
            elif i%8 ==5:
                a2.append(4)
            elif i%8 ==7:
                a2.append(5)

        if (i%10 == 0)|(i%10 ==1):
            a3.append(3)
        elif (i%10 == 2)|(i%10 ==3):
            a3.append(1)
        elif (i%10 == 4)|(i%10 ==5):
            a3.append(2)
        elif (i%10 == 6)|(i%10 ==7):
            a3.append(4)
        elif (i%10 == 8)|(i%10 ==9):
            a3.append(5)
    for i in range(0, num_q, 2):
        a2[i] = 2
    
    for i in range(0, num_q):
        if a1[i] == answers[i]:
            cnt1 += 1
        if a2[i] == answers[i]:
            cnt2 += 1
        if a3[i] == answers[i]:
            cnt3 += 1
    
    if (cnt1 > cnt2)&(cnt1 > cnt3):
        sol = [1]
    elif (cnt2 > cnt1)&(cnt2 > cnt3):
        sol = [2]
    elif (cnt3 > cnt1)&(cnt3 > cnt2):
        sol = [3]
    elif (cnt1 == cnt2)&(cnt1 > cnt3):
        sol = [1,2]
    elif (cnt1 == cnt3)&(cnt1 > cnt2):
        sol = [1,3]
    elif (cnt2 == cnt3)&(cnt2 > cnt1):
        sol = [2,3]
    elif (cnt1 == cnt2)&(cnt1 ==cnt3):
        sol = [1,2,3]
    
    return sol

 

2. 다른 사람 풀이1

def solution(answers):
    pattern1 = [1,2,3,4,5]
    pattern2 = [2,1,2,3,2,4,2,5]
    pattern3 = [3,3,1,1,2,2,4,4,5,5]
    score = [0, 0, 0]
    result = []

    for idx, answer in enumerate(answers):
        if answer == pattern1[idx%len(pattern1)]:
            score[0] += 1
        if answer == pattern2[idx%len(pattern2)]:
            score[1] += 1
        if answer == pattern3[idx%len(pattern3)]:
            score[2] += 1

    for idx, s in enumerate(score):
        if s == max(score):
            result.append(idx+1)

    return result

확실히 깔끔하다.

먼저 각 패턴에 대해서 나는 새로운 리스트(배열)을 만들어야 겠다고 생각했는데 여기서 부터 잘못됐다. 이 분들은 그냥 간단하게 각 pattern을 담고있는 리스트를 만들어서 문제를 푸셨다.

거기에다가 각 정답 갯수를 담고있는 score라는 새로운 리스트를 생성해서 +=를 해주고

 

max함수를 사용해서 간단하게 풀이를 하셨다.

- 이러한 pattern이 주어진 형태의 경우 억지로 해당 리스트를 만들어내야한다는 생각보다 pattern을 담는 리스트를 만들도록 해보자

- list에는 max가 먹는다. 마찬가지로 min도 먹는다. sum도 먹고..

 

3. 다른 사람 풀이2

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):
            if a == v[q % len(v)]:
                s[i] += 1
    return [i + 1 for i, v in enumerate(s) if v == max(s)]

 이 풀이는 위에서 한발 더 진보했다.

이쯤되는 풀이는 과연 시험장에서 한방에 이렇게 써내려갈 수 있을지는 의문.. 연습해야겠지

처음에 pattern을 아예 2차원으로 만들고  for문 두개를 중첩한 다음 return을 comprehension으로 해버리는 

이쯤되면 예술이 아닐까.