codility Fish

最新推荐文章于 2021-03-18 21:26:32 发布

原创最新推荐文章于 2021-03-18 21:26:32 发布 · 441 阅读

0 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

62 篇文章

订阅专栏

codility

44 篇文章

订阅专栏

本文解析了 Codility Lesson 7 中 Fish 问题的解决方案。通过使用栈来跟踪顺流而下的鱼，该算法有效地计算了最终存活的鱼的数量。

Question:codility Lesson7 Fish

My Answer:

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

    for i in range(len(A)):
        if B[i] == 1:
            downstream.append(A[i])
        else:
            while downcnt != 0:
                if downstream[-1] < A[i]:
                    downcnt -= 1
                    downstream.pop()
                else:
                    break
            else:
                cnt += 1
    cnt += len(downstream)
    return cnt