[Algorithm] 연속 펄스 부분 수열의 합 ( Programmers / Python )

문제 설명

어떤 수열의연속 부분 수열에 같은 길이의 펄스 수열을 각 원소끼리 곱하여 연속 펄스 부분 수열을 만들려 합니다. 펄스수열이란 [1, -1, 1, -1 ....] 또는 [-1, 1, -1, 1 ....] 과 같이 1 또는 -1로 시작하면서 1과 -1이 번갈아 나오는 수열입니다.

예를 들어 수열 [2, 3, -6, 1, 3, -1, 2, 4]의 연속 부분 수열 [3, -6, 1]에 펄스 수열 [1, -1, 1]을 곱하면 연속 펄스 부분수열은 [3, 6, 1]이 됩니다. 또 다른 예시로 연속 부분 수열 [3, -1, 2, 4]에 펄스 수열 [-1, 1, -1, 1]을 곱하면 연속 펄스 부분수열은 [-3, -1, -2, 4]이 됩니다.

정수 수열 sequence 가 매개변수로 주어질 때, 연속 펄스 부분 수열의 합 중 가장 큰 것을 return 하도록 solution 함수를 완성해주세요.

 

 

 

---

 

 

 

제한 사항

 

 

 

---

 

 

 

입출력 예

 

 

 

---

 

 

 

문제 풀이

연속 펄스 부분 수열의 합 알고리즘문제는 카데인(Kadane's) 알고리즘을 이용해서 풀은 문제였다.

기존의 방법에서는 O(n^2)의 방법으로 풀 수 밖에 없어서 시간초과가 나왔지만, 카데인 알고리즘을 사용하니 O(n)으로 효율이 개선되었다.

 

python:

 

def solution(sequence):
    sequence_r = []
    pivot = True
    for i in range(len(sequence)):
        if pivot:
            sequence_r.append(sequence[i] * 1)
            pivot = False
        else:
            sequence_r.append(sequence[i] * -1)
            pivot = True
 
    current_max = global_max = sequence_r[0]
    current_max2 = global_max2 = sequence_r[0] * -1
 
    # 카데인 알고리즘
    for r1 in sequence_r[1:]:
        r2 = -1 * r1
 
        current_max = max(r1, current_max + r1)
        current_max2 = max(r2, current_max2 + r2)
 
        global_max = max(global_max, current_max)
        global_max2 = max(global_max2, current_max2)
 
    return max(global_max2, global_max)

 

 

 

 

 

 

 

 

 

 

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/161988

프로그래머스