[COS Pro 1급, Python] 6차 8번 : 지그재그 수열

문제 유형

 빈칸

난이도

 easy

Note 

 1.func_a : check된 배열을 가지고 DP 방식으로 이전 값과 다르면 1 증가시켜서 ret에 저장. 

   func_b : 이전에서 증가하는지 감소하는지 체크하여 ret 배열에 넣음. (0 또는 1이 들어있음. 0 0 1 0 1 0 0 .. 이런식)

   func_c : ret에서 최대값을 출력.

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
INC = 0
DEC = 1

def func_a(arr):
	length = len(arr)
	ret = [0 for _ in range(length)]
	ret[0] = 1
	for i in range(1, length):
		if arr[i] != arr[i-1]:
			ret[i] = ret[i-1] + 1
		else:
			ret[i] = 2
	return ret

def func_b(arr):
	global INC, DEC
	length = len(arr)
	ret = [0 for _ in range(length)]
	ret[0] = -1
	for i in range(1, length):
		if arr[i] > arr[i-1]:
			ret[i] = INC
		elif arr[i] < arr[i-1]:
			ret[i] = DEC
	return ret

def func_c(arr):
	ret = max(arr)
	if ret == 2:
		return 0
	return ret

def solution(S):
	check = func_b(S)
	dp = func_a(check)
	answer = func_c(dp)
	return answer

S1 = [2, 5, 7, 3, 4, 6, 1, 8, 9]
ret1 = solution(S1)

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

S2 = [4, 3, 2, 1, 10, 6, 9, 7, 8]
ret2 = solution(S2)

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

S3 = [1, 2, 3, 4, 5]
ret3 = solution(S3)

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

 

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

※ 반례가 있으면 알려주시면 감사하겠습니다.