[프로그래머스/파이썬] Lv.0 전국 대회 선발 고사 (36')

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

 

풀이 날짜 : 2023-09-21

소요 시간 : 36min

 

---

 

문제 설명

0번부터 n - 1번까지 n명의 학생 중 3명을 선발하는 전국 대회 선발 고사를 보았습니다. 등수가 높은 3명을 선발해야 하지만, 개인 사정으로 전국 대회에 참여하지 못하는 학생들이 있어 참여가 가능한 학생 중 등수가 높은 3명을 선발하기로 했습니다.
각 학생들의 선발 고사 등수를 담은 정수 배열 rank와 전국 대회 참여 가능 여부가 담긴 boolean 배열 attendance가 매개변수로 주어집니다. 전국 대회에 선발된 학생 번호들을 등수가 높은 순서대로 각각 a, b, c번이라고 할 때 10000 × a + 100 × b + c를 return 하는 solution 함수를 작성해 주세요.

 

 

 

나의 풀이

def solution(rank, attendance):
    rank_t = [r for i, r in enumerate(rank) if attendance[i] == True]
    answer = [rank.index(j) for j in sorted(rank_t)][:3]
    return 10000 * answer[0] + 100 * answer[1] + answer[2]

rank_t : 참석한 학생들만 리스트에 모아 놓았다.

answer : 참석한 학생들이 몇 번째 학생인지 모아 놓았다. 여기서 중요한 것은 참석 학생들을 우선 등수별로 지정해야 한다는 것이다. 이 부분 때문에 고민이 오래 걸렸다.

 

 

 

 

다른 풀이

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]

x, i = 학생의 순번