CS/알고리즘_[교재]이것이 취업을 위한 코딩테스트다

[PART2] CH.03_그리디_1_당장 좋은 것만 선택하는 그리디

Jedy_Kim 2021. 6. 8. 15:36
728x90
 현재 상황에서 지금 당장 좋은 것만 고르는 방법 

 


예제3-1

 

  • 거스름돈
당신은 음식점의 계산을 도와주는 점원이다. 카운터에 거스름돈으로 사용할 500원, 100원, 50원, 10원짜리 동전이 무한히 존재한다고 가정한다. 손님에게 거슬러 줘야 할 돈이 N원일 때 거슬러 줘야 할 동전의 최소 개수를 구하라. 단, 거슬러 줘야 할 돈 N은 항상 10의 배수이다. (n=1260일 때)

 

#코드

1
2
3
4
5
6
7
8
9
10
= 1260
count = 0
 
coin_types = [5001005010]
 
for i in coin_types:
    count += n//i    
    n %= i
 
print(count)
cs

 


예제3-2

 

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

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

단, 서로 다른 인덱스에 해당하는 수가 같은 경우에도 서로 다른 것으로 간주한다. 예를 들어 순서대로 3, 4, 3, 4 으로 이루어진 배열이 있을 때 M이 7이고, K가 2라고 가정하자. 이 경우 두 번째 원소에 해당하는 4와 네 번째 원소에 해당하는 4를 번갈아 두 번식 더하는 것이 가능하다. 결과적으로 4 + 4 + 4 + 4 + 4 + 4 + 4인 28이 도출된다. 
배열의 크기 N, 숫자가 더해지는 횟수 M, 그리고 K가 주어질 때 동빈이의 큰 수의 법칙에 따른 결과를 출력하시오.

입력 조건
- 첫 줄에 N(2<=N<=1,000), M(1<=M<=10,000), K(1<=K<=10,000)의 자연수가 주어지며, 각 자연수는 공백으로 구분한다.
- 둘째 줄에 N개의 자연수가 주어진다. 각 자연수는 공백으로 구분한다. 단, 각각의 자연수는 1 이상 10,000 이하의 수로 주어진다.
- 입력으로 주어지는 K는 항상 M보다 작거나 같다.

출력조건
- 첫째 줄에 동빈이의 큰 수의 법칙에 따라 더해진 답을 출력한다.
입력
5 8 3
2 4 5 4 6

출력
46

 

# 단순 풀이

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import sys
 
 
if __name__ == "__main__":
    input   = sys.stdin.readline
    n, m, k = map(int, input().split())
    getArr  = list(map(int, input().split())) 
    getArr.sort()
 
    first  = getArr[-1# 가장 큰 수
    second = getArr[-2# 그 다음 큰 수
    
    result = 0
 
    while True:
        for i in range(k): # 가장 큰 수를 k번 더하기
            if m == 0# m==0 이라면 반복문 탈출
                break
            result += first
            m -= 1
        if m == 0: break
 
        result += second # 두 번째로 큰 수를 한 번 더하기
        m -= 1
    
    print(result)
 
    
cs

 

 

# 시간복잡도를 개선한 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import sys
 
if __name__ == "__main__":
    input   = sys.stdin.readline
    n, m, k = map(int, input().split())
    getArr  = list(map(int, input().split())) 
    getArr.sort()
 
    first  = getArr[-1# 가장 큰 수
    second = getArr[-2# 그 다음 큰 수
    
    result = 0
    '''
    * 이 문제를 풀려면 먼저 반복되는 수열에 대해서 파악해야 한다.
    * 특징 : 가장 큰 수와 두 번째로 큰 수가 더해질 때 특정한 수열 형태로 반복해서 더해지는 특징이 있다.
             ex) 6 6 6 5 6 6 6 5 -> {6, 6, 6, 5} 가 반복된다.
    * 반복되는 수열의 길이 : (k+1)
    * 반복되는 수열의 갯수 : m을 (k+1)로 나눈 몫
    * 가장 큰 수가 등장하는 횟수 : m을 (k+1)로 나눈 몫 곱하기 k
    * 이때 m이 (k+1)로 나누어떨어지지 않는 경우도 고려해야 한다. + m%(k+1)
    '''
    # count : 가장 큰 수의 갯수를 구한다.
    count = m//(k+1)*k
    count += m%(k+1)
 
    result += count*first
    result += (m-count)*second
    
    print(result)
 
cs

예제3-3

 

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

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

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

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

출력 조건
 첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력한다.
입력
3 3
3 1 2
4 1 4
2 2 2
출력
2

입력
2 4
7 3 1 8
3 3 3 4
출력
3

 

# 코드_1 : 모두 직접 구하기

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys
 
if __name__ == "__main__":
    input  = sys.stdin.readline
    n, m   = map(int, input().split())
    getArr = [list(map(int, input().split())) for _ in range(n)]
    minArr = list()
    
    cnt = 0 
    for i in range(len(getArr)):       
        minVal = getArr[i][0
        for j in range(1len(getArr[i])):        
            if minVal > getArr[i][j]:
                minVal = getArr[i][j]
        minArr.append(minVal)
 
    maxVal = 0
    for i in minArr:
        if maxVal < i:
            maxVal = i
    print(maxVal)    
cs

 

#코드_2 : 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import sys
 
if __name__ == "__main__":
    input  = sys.stdin.readline
    n, m   = map(int, input().split())
    
    result = 0
 
    for i in range(n):
        getArr = list(map(int, input().split()))
        # 가장 작은 수 찾기
        min_value = min(getArr)
        # '가장 작은 수'들 중 가장 큰 수 찾기
        result = max(result, min_value)
     
    print(result)
cs

예제3-4

 

  • 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<=100000)과 K(2<=K<=100000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.

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

 

#코드(기본풀이는 생략)--> 이해하기까지 시간이 좀 필요했다...

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import sys
 
if __name__ == "__main__":
    input  = sys.stdin.readline
    n, k   = map(int, input().split())
    
    result = 0
    while True:
        target = (n//k)*k
        result += (n-target)
        n = target     
        if n < k:
            break        
        result += 1
        n //= k
 
    result += (n-1)
    print(result)
cs

 

 

반응형

'CS > 알고리즘_[교재]이것이 취업을 위한 코딩테스트다' 카테고리의 다른 글

[PART2]CH.07_이진탐색  (0) 2021.06.17
[PART2]CH.06_정렬  (0) 2021.06.16
[PART2] CH.05_DFS/BFS  (0) 2021.06.13
[PART2]CH.04_구현  (0) 2021.06.10
목록 및 학습 계획  (0) 2021.06.08