[COS Pro 1급, Python] 1차 9번 : 계단 게임

문제 유형

 한줄 수정

난이도

 easy

Note 

1. 계단은 0칸 부터, 마이너스가 없다. 가위바위보에서 졌다고 계속 내려갈 수 없다. 

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean

def func(record):
	if record == 0:
		return 1
	elif record == 1:
		return 2
	return 0

def solution(recordA, recordB):
	cnt = 0
	for i in range(len(recordA)):
		if recordA[i] == recordB[i]:
			continue
		elif recordA[i] == func(recordB[i]):
			cnt = cnt + 3
		else:
			if cnt > 0:
				cnt = cnt - 1
	return cnt

recordA = [2,0,0,0,0,0,1,1,0,0]
recordB = [0,0,0,0,2,2,0,2,2,2]
ret = solution(recordA, recordB)


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

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.