[Python] 7.2. Fish

app.codility.com/demo/results/trainingWVD5AJ-4CG/

Test results - Codility

---

Task Score

100%

Correctness

100%

Performance

100%

Detected time complexity

O(N)

def solution(A, B):
    
    stack = []
    cnt = 0

    for i in range(len(A)):
        if B[i] == 1:
            stack.append(A[i])
        else:
            while stack:
                f = stack[-1]
                if f > A[i]:
                    break
                else:
                    stack.pop()
            if not stack:
                cnt += 1
    
    return cnt + len(stack)

 

문제를 푸는데 도움이 된 테스트 케이스