프로그래머스 파이썬 기초 트레이닝 :: 주사위 게임 3 (list - count, index, eval+join)
문제 설명
프로그래머스 URL
https://school.programmers.co.kr/learn/courses/30/lessons/181916
프로그래머스 코딩 카테고리
코딩 기초 트레이닝
Day 8 조건문, 문자열 - 주사위 게임 3
Lv.0
문제 내용
1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다. 네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.
네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q)2 점을 얻습니다.
주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.
네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.
제한 사항
a, b, c, d는 1 이상 6 이하의 정수입니다.
문제 이해
[입력] a, b, c, d 주사위 굴렀을때 나오는 숫자
[출력] 5가지 경우에 따라 계산한 점수 값
1. 주사위 4개 값(p)이 모두 동일할때 : 1111 * p
2. 주사위 3개(p)가 동일하고, 1개(q)만 다를때 : (10 * p + q) ** 2
3. 주사위 2개(p)가 동일하고, 다른 2개(q)가 같을때 : (p + q) * abs(p - q)
4. 주사위 2개(p)가 같고, 나머지 2개가 다 다를때(q, r) : q * r
5. 주사위 4개 값이 다 다를때 : min(4개 숫자)
set 함수, count 메서드, index 메서드, max(), min(), abs(), eval() 등을 활용 가능
------------------------
count 리스트 활용
정답 코드
def solution(a, b, c, d):
l = [a,b,c,d]
c = [l.count(x) for x in l]
if max(c) == 4:
return 1111*a
elif max(c) == 3:
return (10*l[c.index(3)]+l[c.index(1)])**2
elif max(c) == 2:
if min(c) == 1:
return eval('*'.join([str(l[i]) for i, x in enumerate(c) if x == 1]))
else:
return (max(l) + min(l)) * abs(max(l) - min(l))
else:
return min(l)리스트를 두개를 사용한다. 주사위 값 리스트와 주사위 횟수를 저장하는 리스트
set 을 이용하면, 주사위 2개에서 동일한 값이나왔는지, 3개에서 동일한 값이 나왔는지 파악이 어렵다.
그래서, set을 쓰면 count도 또 써야하는데, 이렇게 카운트 값 자체를 받고 index 메서드를 사용하면 더 쉬워진다.
① index 메서드
리스트의 값을 넣으면 그 값의 인덱스를 알려주는 역할을 한다. 찾은 인덱스 값은 변수에 저장된다.
문법 : 인덱스 값을 넣을 변수 = list.index(x[, start[, end]])
① x: 찾을 값.
② start (선택): 검색을 시작할 인덱스. 기본값은 0
③ end (선택): 검색을 종료할 인덱스. 기본값은 리스트의 마지막 인덱스
index 함수를 활용해서, 주사위 숫자를 구한다.
l = [a,b,c,d]
c = [l.count(x) for x in l]
if max(c) == 3:
return (10*l[c.index(3)]+l[c.index(1)])**2
다른 사람의 풀이에서 index 함수를 통해서 q, p, r 값을 구한 경우도 있었다. (아래코드)
def solution(a, b, c, d):
nums = [a, b, c, d]
counts = [nums.count(i) for i in nums]
if max(counts) == 4:
return a * 1111
elif max(counts) == 3:
p = nums[counts.index(3)]
q = nums[counts.index(1)]
return (10 * p + q) ** 2
elif max(counts) == 2:
if min(counts) == 2:
return (a + c) * abs(a - c) if a == b else (a + b) * abs(a - b)
else:
p = nums[counts.index(2)]
return (a * b * c * d) / p**2
else:
return min(nums)② eval 함수
l = [a,b,c,d]
c = [l.count(x) for x in l]
if max(c) == 2:
if min(c) == 1:
return eval('*'.join([str(l[i]) for i, x in enumerate(c) if x == 1]))주사위 2개(p)가 같고, 나머지 2개가 다 다를때(q, r) : q * r 를 위와 같이 표현할 수 있다.
댓글
댓글 쓰기