codility CountDistinctSlices

Codility Lesson 15 解题思路

最新推荐文章于 2019-02-19 09:49:18 发布

原创最新推荐文章于 2019-02-19 09:49:18 发布 · 564 阅读

0 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

62 篇文章

订阅专栏

codility

44 篇文章

订阅专栏

本文介绍了一个解决 Codility Lesson 15 CountDistinctSlices 问题的方法。通过使用滑动窗口技术和前缀和优化算法，有效地计算了数组中包含不同整数的切片数量，确保了算法的时间复杂度在 O(N) 范围内。

Question:codility Lesson15 CountDistinctSlices

My Answer:

def solution(M,A):
    front = 0
    back = 0

    lenA = len(A)
    res = 0
    exist = [-1] * (M + 1)


    for front in range(lenA):
        if exist[A[front]] == -1:
            exist[A[front]] = front
        else:
            posback = exist[A[front]] + 1
            res += (posback - back) * (front - back + front - posback + 1)/2
            if res >= 1000000000:
                return 1000000000

            for i in range(back,posback):
                exist[A[i]] = -1

            exist[A[front]] = front

            back = posback

    res += (front - back + 1) * (front - back + 2) / 2

    return min(res,1000000000)