[COS Pro 2급, Python] 6차 4번 : 카드뽑기 게임! (도박은 안되요!)

문제

 

카드를 3장 뽑아 점수를 내는 게임을 하려고 합니다. 각 카드는 색이 칠해져 있고, 숫자가 적혀 있습니다.

획득한 점수를 계산하는 규칙은 다음과 같습니다.

1. 카드 3장의 색이 모두 같다면 획득한 점수는 적힌 숫자의 총합에 3을 곱한 값입니다.
2. 카드 2장의 색이 같고, 1장의 색이 다르다면 획득한 점수는 적힌 숫자의 총합에 2를 곱한 값입니다.
3. 카드 3장의 색이 모두 다르다면 획득한 점수는 적힌 숫자의 총합입니다.

 

뽑은 카드의 색과 숫자를 문자열로 담은 2차원 배열 cards, cards의 길이 cards_len이 매개변수로 주어질 때, 획득한 총 점수를 return 하도록 solution 함수를 작성해주세요.

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

code

def solution(cards):
	answer = 0
    
    #--------------------------------------------
	a = [0, 0, 0]
	for i in cards :
		if i[0] == "blue":
			a[0] += 1
		elif i[0] == "black" :
			a[1] += 1
		elif i[0] == "red" :
			a[2] += 1
		answer = answer + int(i[1])
		
	if 3 in a :
		answer *= 3
	elif 2 in a :
		answer *= 2
        
        #--------------------------------------
	return answer
    
    
cards1 = [["blue", "2"], ["red", "5"], ["black", "3"]]
ret1 = solution(cards1)

print("solution 함수의 반환 값은", ret1, "입니다.")

cards2 = [["blue", "2"], ["blue", "5"], ["black", "3"]]
ret2 = solution(cards2)

print("solution 함수의 반환 값은", ret2, "입니다.")