[이코테] 그리디(Greedy)

해당 글은 동빈나님의 "(이코테 2021 강의 몰아보기) 2. 그리디 & 구현" 영상을 보고, 혼자 공부하기 위한 목적으로 정리한 글입니다. 문제가 있을 시 알려주시면 감사하겠습니다:) 😊

---

 

그리디 알고리즘(Greedy Algorithm)

• 탐욕 알고리즘이라고도 부른다.
• 그 순간에 최적이라고 생각되는 것을 선택하는 방법이다.
• 그리디 해법은 정당성 분석 중요! 단순히 가장 좋아보이는 것을 반복적으로 선택해도 최적의 해를 구할 수 있는 지 검토해야한다.
• 코딩 테스트에서의 대부분의 그리디 문제는 탐욕법으로 얻은 해가 최적의 해가 되는 상황에서, 이를 추론할 수 있어야 풀리도록 출제된다.

 

---

문제: 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번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 작성하세요.

 

난이도: 하 | 시간제한: 2초 | 메모리: 128MB

 

입력 조건 

첫째 줄에 N(1 <= N <= 100,000)과 K(2 <= K <= 100,000)가 공백을 기준으로 하여 각각 자연수로 주어집니다.

출력 조건 

첫째 줄에 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 횟수의 최솟값을 출력합니다.

 

입력 예시

25 5

출력 예시

2

 

---

문제 해결 아이디어

주어진 N에 대하여 최대한 많이 나누기

 

 

내 코드

n, k = map(int, input().split())

count = 0
while(n != 1):
  if n % k == 0:
    n = n // k
  else:
    n -= 1
  count += 1

print(count)

 

클린 코드

n, k = map(int, input().split())

result = 0

while True:
  # N이 K로 나누어 떨어지는 수가 될 때까지 빼기
  target = (n // k) * k
  result += (n - target)
  n = target
  # N이 K보다 작을 때 (더 이상 나눌 수 없을 때) 반복문 탈출
  if n < k:
    break
  # K로 나누기
  result += 1
  n //= k

# 마지막으로 남은 수에 대하여 1씩 빼기
result += (n - 1)
print(result)

 

---

문제: 곱하기 혹은 더하기

각 자리가 숫자(0부터 9)로만 이루어진 문자열 S가 주어졌을 때, 왼쪽부터 오른쪽으로 하나씩 모든 숫자를 확인하며 숫자 사이에 'x' 혹은 '+' 연산자를 넣어 결과적으로 만들어질 수 있는 가장 큰 수를 구하는 프로그램을 작성하세요. 단, +보다 x를 먼저 계산하는 일반적인 방식과는 달리, 모든 연산은 왼쪽에서부터 순서대로 이루어진다고 가정합니다.

 

예를 들어 02984라는 문자열로 만들 수 있는 가장 큰 수는 ((((0 + 2) x 9) x 8) x 4) = 576입니다. 또한 만들어질 수 있는 가장 큰 수는 항상 20억 이하의 정수가 되도록 입력이 주어집니다.

 

난이도: 하 | 시간 제한: 1초 | 메모리 제한: 128MB | 기출: Facebook 인터뷰

 

입력 조건

첫째 줄에 여러 개의 숫자로 구성된 하나의 문자열 S가 주어집니다. (1 <= S의 길이 <= 20)

출력 조건

첫째 줄에 만들어질 수 있는 가장 큰 수를 출력합니다.

 

입력 예시 1

02984

출력 예시 1

576

 

입력 예시 2

567

출력 예시 2

210

 

---

문제 해결 아이디어

두 수에 대하여 연산을 수행할 때,
두 수 중에서 하나라도 1 이하인 경우엔 더하고, 두 수가 모두 2 이상인 경우에는 곱하기

 

 

내 코드

s = input()

result = int(s[0])
for i in range(1, len(s)):
  n = int(s[i])
  if n == 0 or n == 1 or result == 0:
    result += n
  else:
    result *= n

print(result)

 

클린 코드

data = input()

# 첫 번째 문자를 숫자로 변경하여 대입
result = int(data[0])

for i in range(1, len(data)):
  # 두 수 중에서 하나라도 '0' 혹은 '1'인 경우, 곱하기보다는 더하기 수행
  num = int(data[i])
  if num <= 1 or result <= 1:
    result += num
  else:
    result *= num

print(result)

 

---

문제: 모험가 길드

한 마을에 모험가가 N명 있습니다. 모험가 길드에서는 N명의 모험가를 대상으로 '공포도'를 측정했는데, '공포도'가 높은 모험가는 쉽게 공포를 느껴 위험 상황에서 제대로 대처할 능력이 떨어집니다.

 

모험가 길드장인 동빈이는 모험가 그룹을 안전하게 구성하고자 공포도가 X인 모험가는 반드시 X명 이상으로 구성한 모험가 그룹에 참여해야 여행을 떠날 수 있도록 규정했습니다.

 

동빈이는 최대 몇 개의 모험가 그룹을 만들 수 있는지 궁금합니다. N명의 모험가에 대한 정보가 주어졌을 때, 여행을 떠날 수 있는 그룹 수의 최댓값을 구하는 프로그램을 작성하세요.

 

예를 들어 N = 5이고, 각 모험가의 공포도가 다음과 같다고 가정합시다.

2 3 1 2 2

 

이 경우 그룹 1에 공포도가 1, 2, 3인 모험가를 한 명씩 넣고, 그룹 2에 공포도가 2인 남은 두 명을 넣게 되면 총 2개의 그룹을 만들 수 있습니다.

 

또한 몇 명의 모험가는 마을에 그대로 남아 있어도 되기 때문에, 모든 모험가를 특정한 그룹에 넣을 필요는 없습니다.

 

난이도: 하 | 시간 제한: 1초 | 메모리 제한: 128MB | 기출: 핵심 유형

 

입력 조건

- 첫째 줄에 모험가의 수 N이 주어집니다. (1 <= N <= 100,000)
- 둘째 줄에 각 모험가의 공포도의 값을 N 이하의 자연수로 주어지며, 각 자연수는 공백으로 구분합니다.

 

출력 조건

여행을 떠날 수 있는 그룹 수의 최댓값을 출력합니다.

 

입력 예시

5
2 3 1 2 2

 

출력 예시

2

 

---

해결 아이디어

1. 오름차순 정렬 이후에 공포도가 가장 낮은 모험가부터 하나씩 확인

2. '현재 그룹에 포함된 모험가의 수'가 '현재 확인하고 있는 공포도'보다 크거나 같다면 이를 그룹으로 설정

 

 

내 코드

n = int(input())
fear = list(map(int, input().split()))

fear.sort()

group = 0
member = 1
for i in fear:
  if i <= member:
    group += 1
    member = 0
  elif i > member:
    member += 1
  else:
    break

print(group)

 

클린 코드

n = int(input())
data = list(map(int, input().split()))
data.sort()

result = 0 # 총 그룹의 수
count = 0 # 현재 그룹에 포함된 모험가의 수

for i in data: # 공포도를 낮은 것부터 하나씩 확인하기
  count += 1 # 현재 그룹에 해당 모험가를 포함시키기
  if count >= i: # 현재 그룹에 포함된 모험가의 수가 현재의 공포도 이상이라면, 그룹 결성
    result += 1 # 총 그룹의 수 증가시키기
    count = 0 # 현재 그룹에 포함된 모험가의 수 초기화

print(result) # 총 그룹의 수 출력