[Seop's의 코드풀이] 백준12018 Yeonsei TOTO

www.acmicpc.net/problem/12018

12018번: Yonsei TOTO

 

문제요약

가장 많은 과목수를 골라야 한다. 즉 내가 가진 포인트를 최대한 활용해서 많은 강의를 듣는것이 목적인 문제이다. 

 

문제풀이 IDEA

- 각 과목별로 수강할 수 있는 가장 적은 마일리지를 알아낸다.

신청인원이 수강총 정원보다 작을 경우 수강 마일리지는 1

신청인원이 수가총 정원보다 클 경우 수강 마일리지는 다음과 같다. 

수강신청 마일리지를 오름차순으로 정렬 -> [수강총 정원 -1]이 수강 마일리지가 된다. 

- 과목별로 수강 할 수 있는 가장 적은 마일리지를 오름차순으로 정렬한다.

- 가장 작은 것부터 수강한다. 내 마일리지가 0보다 작을 경울 수강을 그만두고 정답을 출력한다. 

 

전체적인 코드는 다음과 같습니다. 질문 및 피드백은 언제나 감사드립니다. 

 

import sys
input = sys.stdin.readline


class Soluction :
    def __init__(self):
        self.N = 0
        self.M = 0
        self.answer = 0
        self.course = []
    
    def firstInput(self) :
        n,m = map(int, input().rstrip().split(" "))
        self.N = n
        self.M = m
        for _ in range(n) :
            self.courseInput()

    def courseInput(self) :
        person, total = map(int, input().rstrip().split(" "))
        arr = list(map(int, input().split(" ")))
        arr = sorted(arr, key = lambda x: -x)
        if person < total :
            self.course.append(1)
        else :
            self.course.append(arr[total-1])
    
    def getAnswer(self) :
        self.course = sorted(self.course)
        for i in self.course :
            if self.M - i >= 0 :
                self.answer += 1 
                self.M -= i
            else :
                break
        return self.answer

if __name__ == "__main__":
    s = Soluction()
    s.firstInput()
    answer = s.getAnswer()
    print(answer)