파이썬 알고리즘-9 주사위 게임

jiffydev·2020년 8월 23일
0

Algorithm

목록 보기
9/134
post-thumbnail

9.주사위 게임

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+21,000 으로 계산되어 12,000원을 받게 된다.3개의 눈이 6, 2, 5로 주어지면 그 중 가장 큰 값이 6이므로 6100으로 계산되어 600원을 상금으로 받게 된다.
N 명이 주사위 게임에 참여하였을 때, 가장 많은 상금을 받은 사람의 상금을 출력하는 프로그램
을 작성하시오.

▣ 입력설명
첫째 줄에는 참여하는 사람 수 N(2<=N<=1,000)이 주어지고 그 다음 줄부터 N개의 줄에 사람
들이 주사위를 던진 3개의 눈이 빈칸을 사이에 두고 각각 주어진다.

▣ 출력설명
첫째 줄에 가장 많은 상금을 받은 사람의 상금을 출력한다.

▣ 입력예제 1
3 3
3 6
2 2 2
6 2 5

▣ 출력예제 1
12000

내 코드

n=int(input())
lst=[]

for i in range(n):
    lst.append(list(map(int, input().split())))
highest=float("-inf")
for j in lst:
    lsts=[0]*7
    for k in j:
        lsts[k]+=1
    
    max_index=lsts.index(max(lsts))
    if lsts[max_index]==2:
        tmp=1000+(max_index)*100
    elif lsts[max_index]==3:
        tmp=10000+(max_index)*1000
    else:
        tmp=max(j)*100
    if tmp>highest:
        highest=tmp

print(highest)

풀긴 풀었는데 효율성은 좋지 않다.
내가 푼 방법은 입력 받은 숫자들을 3개씩 묶어 하나의 리스트에 넣은 뒤 주사위 눈 숫자만큼의 빈 리스트(lsts)를 하나 만든다. 그리고 첫 묶음에 나온 숫자에 해당하는 lsts의 인덱스에 1을 더해줘서 그 숫자가 몇 번 나왔는지 알 수 있게 했다. 이를 바탕으로 3번 나왔을 때, 2번 나왔을 때 1번 나왔을 때를 나눠 각각 조건문으로 계산을 했고, 거기서 나온 금액을 최대라 가정한 뒤 다음 반복문에서 더 큰 금액이 나오면 최댓값을 바꿔주는 방식으로 가장 큰 금액을 구했다.
이걸 반복문 한번에 해결할 수 있으리라고는 생각도 못했다..

풀이

res=0
n=int(input())
for i in range(n):
    tmp=input().split() 
    tmp.sort() 
    a, b, c=map(int, tmp)
    if a==b and b==c:
        money=10000+(a*1000);
    elif a==b or a==c:
        money=1000+(a*100)
    elif b==c:
        money=1000+(b*100)
    else:
        money=c*100
    if money > res:
        res=money

print(res)

반성점

  • unpacking에 대해서는 알고 있었는데 이렇게 사용할 수 있다는 건 몰랐다.

배운 것

  • 비교 개수가 적으면 각각의 변수에 unpack해서 조건문으로 비교해 주는 것으로도 충분하다.
  • 리스트를 여러개 입력해야 한다면 일단 반복문 안에서 한 리스트씩 해결해서 값을 비교해보자.
profile
잘 & 열심히 살고싶은 개발자

0개의 댓글