[이것이 취업을 위한 코딩 테스트다 with 파이썬] 그리디 알고리즘
Updated:
이것이 취업을 위한 코딩 테스트다 - 그리디 알고리즘
#1 큰 수의 법칙
[조건]
-
다양한 수로 이루어진 배열이 있을 때, 주어진 수들을 M번 더하여 가장 큰 수를 만드는 법칙
-
배열의 특정한 인덱스에 해당하는 수가 연속해서 K번 초과하여 더해질 수 없다.
-
서로 다른 인덱스에 해당하는 수가 같은 경우에도 서로 다른 것으로 간주
[입력 조건]
첫째 줄에 N, M, K의 자연수가 주어지며, 각 자연수는 공백으로 구분한다. (2≤ N ≤1,000, 1≤ M ≤10,000, 1≤ K ≤10,000)
둘째 줄에 N개의 자연수가 주어진다. 각 자연수는 공백으로 구분한다.
단, 각각의 자연수는 1이상 10,000이하의 수로 주어진다 입력으로 주어지는 K는 항상 M보다 작거나 같다
[출력 조건]
첫째 줄에 큰 수의 법칙에 따라 더해진 답을 출력한다
N = 5, M = 8, K = 3 일 떄 2, 4, 5, 4, 6 을 입력받으면 결과는 46(= 6 + 6 + 6 + 5 + 6 + 6 + 6 + 5)이 출력된다.
# 코드
N, M, K = map(int, input().split(" "));
numbers = list(input().split(" "));
numbers.sort()
a = int(M//K)
b = int(M%K)
first = int(numbers[-1])
second = int(numbers[-2])
result = (first * a * K) + (second * b)
print(result)
#2 숫자 카드 게임
[조건]
-
숫자가 쓰인 카드들이 N x M 형태로 놓여 있다. 이때 N은 행의 개수를 의미하며, M은 열의 개수를 의미한다.
-
먼저 뽑고자 하는 카드가 포함되어 있는 행을 선택한다.
-
선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑아야 한다.
-
따라서 처음에 카드를 골라낼 행을 선택할 때, 이후에 해당 행에서 가장 숫자가 낮은 카드를 뽑를 것을 고려하여 최종적으로 가장 높은 숫자의 카드를 뽑을 수 있도록 전략을 세워야 한다.
[입력 조건]
첫째 줄에 숫자 카드들이 놓인 행의 개수 N과 열의 개수 M이 공백을 기준으로 하여 각각 자연수로 주어진다. (1 <= N,M<= 100)
둘째 줄부터 N개의 줄에 걸쳐 각 카드에 적힌 숫자가 주어진다.
각 숫자는 1 이상 10,000 이하의 자연수이다.
[출력 조건]
첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력한다.
# 코드
n, m = map(int, input().split(" "))
table = []
a = 0
for i in range(n):
items = list(input().split(" "))
table.append(items)
for i in range(1,len(table)):
if min(table[i-1]) < min(table[i]):
a = i
print(min(table[a]))
#3 1이 될 때 까지
[조건]
어떠한 수가 1이 될 때까지 아래의 두 과정 중 하나를 택해 반복적으로 수행하려고 한다.
-
N에서 1을 뺀다.
-
N을 K로 나눈다.
N과 K가 주어질 때 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 작성하시오.
[입력 조건]
첫째 줄에 N(2<= N <= 100,000>과 K(2<= K <= 100,000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.
[출력 조건]
첫째 줄에 N이 1이 될 때까지 1번 혹은 2번으 과정을 수행해야 하는 획수의 최솟값을 출력한다.
# 코드
n, k = map(int, input().split(" "))
count = 0
while n != 1:
if n % k == 0:
n = n // k
count += 1
else:
n -= 1
count += 1
print(count)