[알고리즘]최대 점수 구하기

김도연·2024년 2월 4일

알고리즘

목록 보기
45/56

문제

이번 정보올림피아드대회에서 좋은 성적을 내기 위하여 현수는 선생님이 주신 N개의 문제를 풀려고 합니다. 각 문제는 그것을 풀었을 때 얻는 점수와 푸는데 걸리는 시간이 주어지게 됩 니다. 제한시간 M안에 N개의 문제 중 최대점수를 얻을 수 있도록 해야 합니다. (해당문제는 해당시간이 걸리면 푸는 걸로 간주한다, 한 유형당 한개만 풀 수 있습니다.)

▣ 입력설명
첫 번째 줄에 문제의 개수N(1<=N<=20)과 제한 시간 M(10<=M<=300)이 주어집니다.
두 번째 줄부터 N줄에 걸쳐 문제를 풀었을 때의 점수와 푸는데 걸리는 시간이 주어집니다.

▣ 출력설명
첫 번째 줄에 제한 시간안에 얻을 수 있는 최대 점수를 출력합니다.

입력예제1

5 20
10 5
25 12
15 8
6 3
7 4

출력예제1

41

[내 코드]

def DFS(L,time_sum,score_sum,s):
    global max_score
    if time_sum>m:
        return
    if time_sum<=m:
        if score_sum>max_score:
            max_score=score_sum
        for i in range(s,n):
            DFS(L+1,time[i]+time_sum,score_sum+score[i],i+1)

if __name__=="__main__":
    n,m=map(int,input().split())
    time=[0]*n
    score=[0]*n
    for i in range(n):
        score[i],time[i]=map(int,input().split())
    max_score=-217000000
    DFS(0,0,0,0)
    print(max_score)

1.DFS에 변수를 총 4개로 구성 time의 합이 입력값 m보다 크면 가지치기.
2.score_sum>max_score 조건을 통해서 계속해서 최댓값을 업데이트 해준다.
3. 중복되지않도록 조합방식으로 문제를 푼 점수와 시간을 score_sum,time_sum에 저장해가면서 DFS를 업데이트.
4.DFS모두 완료 후 max_score를 출력

[해설코드]

def DFS(L,sum,time):
	global res
    if time>m:
    	return
	if L==n:
    	if sum>res:
        	res=sum
     else:
     	DFS(L+1,sum+pv[L],time+pt[L]) //문제를 풀었을 경우
        DFS(L+1,sum,time)//문제를 풀지 않았을 경우
            
    
    
if __name__=="__main__":
	n,m=map(int,input().split())
    pv=list()
    pt=list()
    for i in range(n):
    	a,b=map(int,input().split())
        pv.append(a)
        pt.append(b)
    res=-2147000000
    DFS(0,0,0)
    print(res)
  1. 부분집합과 같은 알고리즘
  2. 하나의 문제마다 문제를 해결하는지 해결 못 하는지 경우를 나눈다.

0개의 댓글