[오늘부터 코딩테스트]파이썬으로 프로그래머스 코딩 기초 프로그램 풀기

[파이썬] 프로그래머스 Lv0. 뒤에서 5 등위로

 

안녕하세요 머킹입니다.

버전 충돌이 너무 많아서 고민이 되네요...

쓰고 싶은 모델을 코랩으로 돌리려고 했는데 무조건 버전 충돌이 있어서

로컬로 돌려야 할 것 같은데 그러면 어마무시한 시간이 걸릴 것 같아서..

어떻게 해야 좋을지 되게 고민이 많습니다.

 

---

뒤에서 5등 위로

 

문제 설명

정수로 이루어진 리스트 num_list가 주어집니다. num_list에서 가장 작은 5개의 수를 제외한 수들을 오름차순으로 담은 리스트를 return하도록 solution 함수를 완성해 주세요.

 

def solution(num_list):
    num_list.sort()
    return num_list[5:]

 

저번 글 마지막 문제와 엄청 유사하죠?

5가 앞이냐 뒤냐의 차이입니다 ㅎㅎ

 

---

전국 대회 선발 고사

 

문제 설명

0번부터 n - 1번까지 n명의 학생 중 3명을 선발하는 전국 대회 선발 고사를 보았습니다.

등수가 높은 3명을 선발해야 하지만, 개인 사정으로 전국 대회에 참여하지 못하는 학생들이 있어 참여가 가능한 학생 중 등수가 높은 3명을 선발하기로 했습니다.

각 학생들의 선발 고사 등수를 담은 정수 배열 rank와 전국 대회 참여 가능 여부가 담긴 boolean 배열 attendance가 매개변수로 주어집니다. 전국 대회에 선발된 학생 번호들을 등수가 높은 순서대로 각각 a, b, c번이라고 할 때 10000 × a + 100 × b + c를 return 하는 solution 함수를 작성해 주세요.

 

def solution(rank, attendance):
    arr = sorted([(x, i) for i, x in enumerate(rank) if attendance[i]])
    return arr[0][1] * 10000 + arr[1][1] * 100 + arr[2][1]

 

생각보다 너무 어려워서

여러 가지를 시도해 봤습니다.

 

저렇게 푸는 방법이 있어서 놀라웠어요.

def solution(rank, attendance):
    array = [(rank[i], i) for i in range(len(rank))]
    array.sort(key=lambda x: x[0])
    pick = [0, 0, 0]
    index = 0
    count = 0
    for i in range(len(array)):
        if attendance[array[i][1]]:
            pick[index] = array[i][1]
            index += 1
            count += 1
        if count == 3:
            break
    answer = 10000 * pick[0] + 100 * pick[1] + pick[2]
    return answer

저의 엄청난 복잡 코드를 보니 또 한 번 느끼네요..