문제설명
수열 S가 주어질 때, 이 수열의 연속된 부분 수열 중 지그재그 수열 길이의 최댓값을 구하려 합니다.
지그재그 수열이란 첫 번째 원소부터 인접한 원소의 차이가 증가 → 감소 → 증가 → 감소 ... 혹은 감소 → 증가 → 감소 → 증가 ... 순으로 나타나는 수열을 말합니다. 단, 수열의 길이는 3 이상이어야 합니다.
예를 들어 수열이 [ 2, 5, 7, 3, 4, 6, 1, 8, 9]인 경우, 연속된 부분 수열 [5, 7, 3, 4]가 부분 수열 중 가장 긴 지그재그 수열이 됩니다.
부분 수열 중 가장 긴 지그재그 수열의 길이를 구하기 위해 다음과 같이 프로그램 구조를 작성했습니다.
~~~
1. 각 숫자가 바로 이전 숫자보다 증가했는지, 혹은 감소했는지 표시한 리스트를 만듭니다.
1-1. "증가"는 "INC", "감소"는 "DEC"로 표시합니다.
1-2. 첫 번째 숫자는 증가도, 감소도 하지 않았다는 의미에서 -1로 표시합니다.
2. 1단계에서 만든 증가, 감소 리스트를 이용해 각 숫자를 마지막으로 하는 지그재그 수열 중 가장 긴 것의 길이를 담은 리스트를 만듭니다.
2-1. 바로 전 숫자가 "증가"이고 현재 숫자가 "감소"이거나, 전 숫자가 "감소"이고 현재 숫자가 "증가"이면,
현재 숫자를 마지막으로 하는 지그재그 수열 중 가장 긴 것의 길이는 (바로 전 숫자를 마지막으로 하는 지그재그 수열중 가장 긴 것의 길이 + 1)입니다.
2-2. 그렇지 않으면 현재 숫자를 마지막으로 하는 지그재그 수열 중 가장 긴 것의 길이는 2입니다.
2-3. 단, 첫 번째 숫자의 길이는 1로 초기화합니다.
3. 2단계에서 구한 리스트에 담긴 값 중 최댓값이 부분 수열 중 가장 긴 지그재그 수열의 길이입니다.
3-1. 만약 최댓값이 2라면 0을 return 합니다.
3-2. 그 외에는 최댓값을 return 합니다.
~~~
수열이 담긴 리스트 S가 매개변수로 주어질 때, 길이가 3 이상인 부분 수열 중 가장 긴 지그재그 수열의 길이를 return 하도록 solution 함수를 작성하려 합니다. 위 구조를 참고하여 코드가 올바르게 동작할 수 있도록 빈칸에 주어진 func_a, func_b, func_c 함수와 매개변수를 알맞게 채워주세요.
매개변수 설명
수열이 담긴 리스트 S가 solution 함수의 매개변수로 주어집니다.
* S의 길이는 3 이상 100 이하입니다.
* S의 원소는 1 이상 100 이하인 자연수이며, 같은 숫자가 중복해서 나타나지 않습니다.
return 값 설명
길이가 3 이상인 부분 수열 중 가장 긴 지그재그 수열의 길이를 return 해주세요.
* 만약 지그재그 수열이 없다면 0을 return 해주세요.
예시
| S | return |
|------------------------------|--------|
| [2, 5, 7, 3, 4, 6, 1, 8, 9] | 4 |
| [4, 3, 2, 1, 10, 6, 9, 7, 8] | 7 |
| [1, 2, 3, 4, 5] | 0 |
예시 설명
예제 #1
문제 예시와 같습니다.
예제 #2
[2, 1, 10, 6, 9, 7, 8]이 부분 수열 중 가장 긴 지그재그 수열입니다.
예제 #3
부분 수열중 지그재그 수열이 없습니다.
문제 코드
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_@@@(@@@)
dp = func_@@@(@@@)
answer = func_@@@(@@@)
return answer풀이
INC = 0
DEC = 1
# 2. 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
# 1. 각 숫자가 바로 이전 숫자보다 증가했는지, 혹은 감소했는지 표시한 리스트를 만듭니다.
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
# 3. 2단계에서 구한 리스트에 담긴 값 중 최댓값이 부분 수열 중 가장 긴 지그재그 수열의 길이입니다.
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
댓글