[이코테]그리디 문제 풀이(큰 수의 법칙, 숫자 카드 게임)

큰 수의 법칙

'큰 수의 법칙'은 일반저긍로 통계 분야에서 다루어지는 내용이지만 동빈이는 본인만의 방식으로 다르게 사용하고 있다. 동빈이의 큰 수의 법칙은 다양한 수로 이루어진 배열이 있을 때 주어진 수들을 M번 더하여 가장 큰 수를 만드는 법칙이다. 단, 배열의 특정한 인덱스(번호)에 해당하는 수가 연속해서 K번을 초과하여 더해질 수 없는 것이 이 법칙의 특징이다.

예를 들어 순서대로 2, 4, 5, 4, 6으로 이루어진 배열이 있을 때 M이 8이고, K가 3이라고 가정하자. 이 경우 특정한 인덱스의 수가 연속해서 세 번까지만 더해질 수 있으므로 큰 수의 법칙에 따른 결과는 6+6+6+5+6+6+6+5인 46이 된다.

단, 서로 다른 인덱스에 해당하는 수가 같은 경우에도 서로 다른 것으로 간주한다. 예를 들어 순서대로 3, 4, 3, 4, 3으로 이루어진 배열이 있을 때 M이 7이고, K가 2라고 가정하자. 이 경우 두 번째 원소에 해당하는 4와 네 번째 원소에 해당하는 4를 번갈아 두 번씩 더하는 것이 가능하다. 결과적으로 4+4+4+4+4+4+4인 28이 도출된다.

배열의 크기 N, 숫자가 더해지는 횟수 M, 그리고 K가 주어질 때 동빈이의 큰 수의 법칙에 따른 결과를 출력하시오.

 

내 답안

n, m, k = map(int, input().split())
arr = list(map(int, input().split()))
sum = 0

arr.sort(reverse=True)   #내림차순 정렬
a, b = arr[0], arr[1]   #a가 제일 큰 값, b가 그 다음으로 큰 값

if a == b:   #둘이 같은 값이라면 M번 더하기 수행
  for _ in range(m):
    sum += a
else:   #둘이 다른 값이라면
  while m > 0:   #m이 0이거나 0보다 작으면 멈추기
    if m - k == 0:   
      break
    elif m - k < 0:   #m<k이면 k-m만큼 더하기
      c = k-m
    else:
      c = k

    for _ in range(c):
        sum += a
    sum += b
    m -= (k+1)   #a를 k번, b를 1번 총(k+1)번 수행
  
print(sum)

a와 b가 같을 때와 다를 때를 나눠서 생각하여 코드를 설계했으나, 답안 예시를 보니 그렇게 할 필요가 없었다...

 

답안 예시

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

data.sort()
first = data[n-1]
second = data[n-2]

result = 0

while True:
  for i in range(k):   #가장 큰 수를 k번 더하기
    if m == 0:
      break
    result += first
    m -= 1   #더할 때마다 -1 계산
  if m == 0:   #m이 0이라면 반복문 탈출
    break
  result += second   #두번째로 큰 수는 1번만 더하기
  m -= 1   #더했으니 -1 계산

print(result)

가장 큰 수를 k번 더하고 두번째 수를 1번 더하는 과정을 반복한다.

 

수학적 아이디어를 이용한 추가 예시

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

data.sort()
first = data[n-1]
second = data[n-2]

#가장 큰 수가 더해지는 횟수 계산
count = int(m/(k+1))*k
count += m%(k+1)

result = 0
result += (count)*first   #가장 큰 수 더하기
result += (m-count)*second   #두 번째로 큰 수 더하기

print(result)

2, 4, 5, 4, 6으로 이루어진 배열이 있을 때 M이 8이고, K가 3일 때

6+6+6+5가 2번 반복되는 것을 알 수 있다.

즉, 반복되는 수열에 대해서 파악하여 문제를 해결할 수 있다. 가장 큰 수와 두 번째로 큰 수가 더해질 때는 특정한 수열 형태로 일정하게 반복해서 더해지는 특징이 있다.

---

숫자 카드 게임

숫자 카드 게임은 여러 개의 숫자 카드 중에서 가장 높은 숫자가 쓰인 카드 한 장을 뽑는 게임이다. 단, 게임의 룰을 지키며 카드를 뽑아야 하고 룰은 다음과 같다.

1. 숫자가 쓰인 카드들이 N x M 형태로 놓여 있다. 이때 N은 행의 개수를 의미하며, M은 열의 개수를 의미한다.

2. 먼저 뽑고자 하는 카드가 포함되어 있는 행을 선택한다.

3. 그다음 선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑아야 한다.

4. 따라서 처음에 카드를 골라낼 행을 선택할 때, 이후에 해당 행에서 가장 숫자가 낮은 카드를 뽑을 것을 고려하여 최종적으로 가장 높은 숫자의 카드를 뽑을 수 있도록 전략을 세워야 한다.

예를 들어 3x3 형태로 카드들이 놓여있다고 가정하자.

여기서 카드를 골라낼 행을 고를 때 첫 번째 혹은 두번째 행을 선택하는 경우, 최종적으로 뽑는 카드는 1이다. 하지만 세 번째 행을 선택하는 경우 최종적으로 뽑는 카드는 2이다. 따라서 이 예제에서는 세 번째 행을 선택하여 숫자가 2가 쓰여진 카드를 뽑는 것이 정답이다.

카드들이 N x M형태로 놓여있을 때, 게임의 룰에 맞게 카드를 뽑는 프로그램을 만드시오.

 

내 답안

n, m = map(int, input().split())
arr = []   #입력받을 배열
res = []   #각 행의 최소값

for _ in range(n):
  a = list(map(int, input().split()))
  arr.append(a)

for i in range(n):
  res.append(min(arr[i]))

print(max(res))

 

 

답안 예시

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

result = 0

for i in range(n):
  data = list(map(int, input().split()))
  min_value = min(data)
  result = max(result, min_value)

print(result)

입력을 받으면서 그때 그때 확인하고 비교

max(a, b)를 하면 a와 b를 비교하여 큰 값을 return

 

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

result = 0

for i in range(n):
  data = list(map(int, input().split()))
  min_value = 1001
  for a in data:
    min_value = min(min_value, a)
  result = max(result, min_value)

print(result)

반복문을 통해 1001이라는 큰 값과 비교하여 최소값을 구한 뒤 최대값을 구한다.

---

추가적으로 2차원 배열 입력받는 방법

arr = []

for i in range(n):    
	arr.append(list(map(int, input().split())))

arr = [list(map(int, input().split())) for _ in range(n)]

 

 

 

<출처>

이것이 취업을 위한 코딩테스트다 with 파이썬

https://blockdmask.tistory.com/411

https://dailylifeofdeveloper.tistory.com/132