Lesson 10: Prime and composite numbers → Peeks

문제

Peaks coding task - Learn to Code - Codility

---

문제 풀이

각 구간을 나눌 때, 각 구간마다 peek를 포함하여야 한다. 문제를 해결하기 위해 peek인 구간을 찾는다.

나눌 수 있는 구간은 주어진 A 리스트의 길이의 약수를 의미하는 것이므로, 약수에 대한 구간을 확인하고 peek를 포함하는 지 확인한다.

 

---

코드

def solution(A):
    peeks = []
    length = len(A)
    for i in range(1, length - 1):
        if A[i - 1] < A[i] > A[i + 1]:
            peeks.append(i)

    cnt = len(peeks)
    if not cnt:
        return 0

    for i in range(cnt, 0, -1):
        if length % i == 0:
            block_size = length // i
            check = [False] * i
            for peek in peeks:
                idx = peek // block_size
                if not check[idx]:
                    check[idx] = True

            if all(check):
                return i
    return 0