[Python/BOJ] 16501. 만족도 점수

https://www.acmicpc.net/problem/16501

16501번: 만족도 점수

문제

테니스 동호회 회장은 매주 참가 회원들이 만족할 만 하도록 2대 2 복식 조들을 짜야 한다. 각 회원은 참여한 게임이 대등하게 펼쳐졌을 수록 만족도가 높다. 참가 회원들의 실력 점수는 0 이상 10이하의 정수로 주어진다고 가정할 때, 한 경기에 참여한 회원의 만족도 점수는 다음과 같이 표현된다.

1 -  ( |상대 팀의 실력 점수 평균 - 본인 팀의 실력 점수 평균| / 10)

이 점수는 최악의 경우 0, 최상의 경우 1점을 범위로 갖는다. 

회장의 목표는 너무 불만족해 탈퇴하는 회원이 없도록 하는 것이다. 이를 위해 모든 회원들이 최소 1번은 참가하게 하고, 만족도 점수의 하한을 극대화 하고 싶다.

2개의 테니스 코트를 쓸 수 있고, 각 코트에서 한 게임씩만 할 수 있으며 8명의 회원이 참여한다고 하자. 목표에 맞게 조를 짰을 때, 만족도 점수의 하한을 구하는 프로그램을 작성하라.

입력

8명 회원의 실력 점수가 입력으로 주어진다. 점수는 공백으로 구분되어져 있다.

출력

첫째 줄에 만족도 점수의 하한을 출력한다.

정답이 정수 또는 소수점 한 자리의 실수로 표현할 수 있는 경우 소수점 첫째 자리까지, 두 자리의 실수로 표현할 수 있는 경우 소수점 둘째 자리까지 출력한다.

풀이

Code

import sys
from itertools import combinations
input = sys.stdin.readline

# 1. 코트의 만족도 점수 계산 함수 생성
def return_score(people) :
    # 1-1. 인원의 실력 점수 리스트 생성 후 정렬
    array = []
    for idx in people :
        array.append(ability[idx])
    array.sort()
    # 1-2. 팀을 나눠 만족도 점수 계산
    score = 1 - (abs((array[0] + array[3])/2 - (array[1] + array[2])/2)/10)
    # 1-3. 계산된 점수 리턴
    return score
def solution(ability) :
    ans = -float('inf')
    # 2. 1번 코트에 들어갈 인원 조합 구하기
    combination = combinations(range(1, 9), 4)
    for court1 in combination :
        # 3. 2번 코트에 들어갈 인원 조합 구하기
        court2 = [i for i in range(1, 9) if i not in court1]
        # 4. 두 코트 중 낮은 만족도 구하기
        # 5. 만족도의 최댓값 업데이트
        ans = max(ans, min(return_score(court1), return_score(court2)))
    # 6. 결과 출력
    print(round(ans, 2))
    
if __name__ == '__main__' :
    ability = [0] + list(map(float, input().split()))
    solution(ability)