[PCCP 모의고사 1] 2번 - Python

https://school.programmers.co.kr/learn/courses/15008/15008-pccp-%EB%AA%A8%EC%9D%98%EA%B3%A0%EC%82%AC-1%ED%9A%8C

PCCP 모의고사 1회

 

처음 딱 보았을때 itertools 를 쓰면 된다는 생각은 들었는데, 세부적인 내용까지 생각하기 은근 까다로운 문제였다.

정석적인 깊이/넓이 탐색으로도 되겠다 란 생각은 하였는데, 나중에 한번 그 방식으로도 해볼 예정이다.

내가 작성한 코드는 생각보다 세련되지않아서 다른 풀이의 코드를 기재한다.

 

다른 인원 코드 

from itertools import permutations

def solution(ability):
    answer = 0
    student_cnt = len(ability) # 학생 수
    event_cnt = len(ability[0]) # 종목 수

    students = [i for i in range(student_cnt)] # list(range(student_cnt)) 해도 동일
    for perm in list(permutations(students, event_cnt)):
        temp = 0
        for i, p in enumerate(perm):
            temp += ability[p][i]
        if temp > answer:
            answer = temp
    return answer