얼레벌레

[Baekjoon] 주사위 세개 본문

Coding Test

[Baekjoon] 주사위 세개

낭낭이 2022. 2. 16. 20:27

문제

1에서부터 6까지의 눈을 가진 3개의 주사위를 던져서 다음과 같은 규칙에 따라 상금을 받는 게임이 있다. 

  1. 같은 눈이 3개가 나오면 10,000원+(같은 눈)×1,000원의 상금을 받게 된다. 
  2. 같은 눈이 2개만 나오는 경우에는 1,000원+(같은 눈)×100원의 상금을 받게 된다. 
  3. 모두 다른 눈이 나오는 경우에는 (그 중 가장 큰 눈)×100원의 상금을 받게 된다.  

예를 들어, 3개의 눈 3, 3, 6이 주어지면 상금은 1,000+3×100으로 계산되어 1,300원을 받게 된다. 또 3개의 눈이 2, 2, 2로 주어지면 10,000+2×1,000 으로 계산되어 12,000원을 받게 된다. 3개의 눈이 6, 2, 5로 주어지면 그중 가장 큰 값이 6이므로 6×100으로 계산되어 600원을 상금으로 받게 된다.

3개 주사위의 나온 눈이 주어질 때, 상금을 계산하는 프로그램을 작성 하시오.

입력

첫째 줄에 3개의 눈이 빈칸을 사이에 두고 각각 주어진다. 

출력

첫째 줄에 게임의 상금을 출력 한다.

예제 입력 1 

3 3 6

예제 출력 1 

1300

예제 입력 2 

2 2 2

예제 출력 2 

12000

예제 입력 3 

6 2 5

예제 출력 3 

600

✔ 배운점

  • 자료형(딕셔너리) 복습 및 관련 함수
  • operator 패키지, itemgetter 모듈
  • collection패키지
  • set, collections.Counter(x) 함수들이 무조건 정렬된다는 보장 X
    • 처음엔 set()으로 오름차순 정렬이 된다고 생각하여 했는데 틀렸었다. set, collections.Counter 둘 다 정렬이 보장된 함수는 아니라는 점!
# 첫번째 시도
import operator, collections
N = list(map(int,input().split()))
dic = collections.Counter(N)
l = len(set(N))
if l == 1:
    print(10000+list(dic.keys())[0]*1000)
elif l == 3:
    print(max(list(dic.keys()))*100)
else:
    k =sorted(dic.items(),key=operator.itemgetter(1),reverse = True)[0][0]
    print(1000+k*100)

# 최종
N = list(map(int,input().split()))
l = len(set(N))
if l == 1:
    print(10000+N[0]*1000)
elif l == 3:
    print(max(N)*100)
else:
    print(1000+[i for i in N if N.count(i) == 2][0]*100)

# 최종 input => sys.stdin.readline
import sys
N = list(map(int,sys.stdin.readline().split()))
l = len(set(N))
if l == 1:
    print(10000+N[0]*1000)
elif l == 3:
    print(max(N)*100)
else:
    print(1000+[i for i in N if N.count(i) == 2][0]*100)

패키지를 불러와 패키지 내의 모듈을 사용하면 메모리 사용도 많아지고 시간복잡도도 커진다. collections와 operator 패키지를 알았지만 정말 필요할 때만 쓰고 구조를 생각해 함수를 내가 직접 구현하는 쪽으로 생각해야겠다.

그래서 메모리를 절약하고 시간을 단축하기 위해 패키지를 사용하는 경우의 코드는 없앴더니 메모리 절약, 시간절약 모두 성공했다

마지막으로 input이 아닌 sys.stdin.readline으로 코드를 바꿨더니 4ms가 줄어들었다.

🙄 input을 쓸 때 더 시간이 줄어드는 경우가 있고 sys.stdin.readline을 쓸 때 시간이 더 줄어드는 경우가 있는데 차이점이 뭘지 궁금하다.

'Coding Test' 카테고리의 다른 글

[Baekjoon] 상수  (0) 2022.02.17
[Baekjoon] 알파벳 찾기  (0) 2022.02.17
[Baekjoon] 한수  (0) 2022.02.15
[Baekjoon] OX퀴즈  (0) 2022.02.14
[Programmers/Python3] 숫자 문자열과 영단어  (0) 2022.02.08
Comments