Greedy

Vincent·2023년 4월 5일

현재 상황에서 지금 당장 좋은 것만 고르는 방법
현재의 선택이 나중에 미칠 영향에 대해서는 고려하지 않음
보통 코테의 그리디 알고리즘 유형의 문제는 창의력을 요구한다
문제 풀이를 위한 최소한의 아이디어를 떠올리고 이것이 정당한지 검토할 수 있어야 답을 도출할 수 있다.
문제에서 ''가장 큰 순서대로'', ''가장 작은 순서대로''와 같은 기준을 알게 모르게 제시해준다. ex) 거스름돈
따라서 정렬 알고리즘과 자주 짝을 이뤄 출제됨

예제1 - 큰 수의 법칙

문제 설명

다양한 수로 이루어진 배열이 있을 때 주어진 수들을 M번 더하여 가장 큰 수를 만드는 법칙.
단 배열의 특정한 인덱스에 해당하는 수가 연속해서 K번을 초과하여 더해질 수 없다.

제한사항

첫째 줄어 N(2 <= N <= 1000), M( 1 <= M <= 10,000), K(1 <= K <= 10,000)의 자연수가 주어지며, 각 자연수는 공백으로 구분한다.
둘째 줄에 N개의 자연수가 주어진다. 각 자연수는 공백으로 구분한다. 단, 각각의 자연수는 1 이상 10,000이하의 수로 주어진다.
입력으로 주어지는 K는 항상 M보다 작거나 같다.
첫째 줄에 동빈이의 큰 수의 법칙에 따라 더해진 답을 출력한다.

풀이

#큰 수의 법칙
import sys

N, M, K = map(int, sys.stdin.readline().split()) #배열 크기, 숫자가 더해지는 횟수, 연속해서 더할 수 있는 횟수

arr = list(map(int, sys.stdin.readline().split()))

arr = sorted(arr, reverse=True)

ans = 0
cnt = 0 #더한 횟수
while cnt != M:
	#더하는 게 처음이 아니고 제일 큰 수를 최대한으로 더할 수 있는 횟수에 도달했을 때
    if cnt != 0 and cnt % K == 0:
        ans += arr[1]
        cnt == 0 #cnt 갱신
    else:
        ans += arr[0]
    cnt += 1

print(ans)

예제2 - 숫자 카드 게임

문제 설명

숫자 카드 게임은 여러 개의 숫자 카드 중에서 가장 높은 숫자자가 쓰인 카드 한 장을 뽑는 게임이다. 단, 게임의 룰을 지키며 카드를 뽑아야 하고 룰은 다음과 같다.
1. 숫자가 쓰인 카드들이 N X M 형태로 놓여 있다. 이때 N은 행의 개수를 의미하며, M은 열의 개수를 의미한다.
2. 먼저 뽑고자 하는 카드가 포함되어 있는 행을 선택한다.
3. 그다음 선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑아야 한다.
4. 따라서 처음에 카드를 골라낼 행을 선택할 때, 이후에 해당 행에서 가장 숫자가 낮은 카드를 뽑을 것을 고려하여 최종적으로 가장 높은 숫자의 카드를 뽑을 수 있도록 전략을 세워야 한다.
카드들이 N X M 형태로 놓여 있을 때, 게임의 룰에 맞게 카드를 뽑는 프로그램을 만드시오.

제한사항

첫째 줄에 숫자 카드들이 놓인 행의 개수 N과 열의 개수 M이 공백을 기준으로 하여 각각 자연수로 주어진다. (1<=N,M<=100)
둘째 줄부터 N개의 줄에 걸쳐 각 카드에 적힌 숫자가 주어진다. 각 숫자는 1 이상 10,000 이하의 자연수이다.
첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력한다.

풀이

#숫자 카드 게임
#각 행마다 가장 작은 수를 찾은 뒤에 그 수 중에서 가장 큰수를 찾는 아이디어
import sys
N, M = map(int, sys.stdin.readline().split())
arr = [list(map(int, sys.stdin.readline().split())) for i in range(N)]

for i in range(N):
    arr[i] = sorted(arr[i])

maxV = 0
for i in range(N):
    if maxV < arr[i][0]:
        maxV = arr[i][0]

print(maxV)

예제3 - 1이 될때까지

문제 설명

어떠한 수 N이 1이 될 때까지 다음의 두 과정 중 하나를 반복적으로 선택하여 수행하려고 한다. 단, 두번째 연산은 N이 K로 나누어떨어질 때만 선택할 수 있다.
1. N에서 1을 뺀다.
2. N을 K로 나눈다.
예를 들어 N이 17, K가 4라고 가정하자. 이때 1번의 과정을 한 번 수행하면 N은 16이 된다. 이후에 2번의 과정을 두 번 수행하면 N은 1이 된다. 결과적으로 이 경우 전체 과정을 실행한 횟수는 3이 된다. 이는 N을 1로 만드는 최소 횟수이다.
N과 K가 주어질 때 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 작성하시오.

제한사항

첫째 줄에 N(2≤ N ≤ 100,000)과 K(2≤ K ≤ 100,000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.
첫째 줄에 N이 1이 될때까지 1번 혹은 2번의 과정을 수행해야 하는 횟수의 최솟값을 출력한다.

풀이

#1이 될때까지
#최대한 많이 나누기

import sys
N, K = map(int, sys.stdin.readline().split())

cnt = 0
while N != 1:
    if N % K != 0: #N이 K로 안나누어 떨어질 때
        N -= 1 #1번 선택
        cnt += 1
    else: 
        N /= K
        cnt += 1

print(cnt)