【kickstart 2020 round C】第三题 Perfect Subarray Python 解题思路（test 2 TLE）

最新推荐文章于 2022-04-16 21:06:32 发布

原创最新推荐文章于 2022-04-16 21:06:32 发布 · 419 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法

kickstart 专栏收录该内容

21 篇文章

订阅专栏

本文探讨了在解决复杂算法问题时，Python与C++的性能差异。通过PerfectSubarray题目，作者分析了使用Python实现的算法在特定测试案例中遭遇的时间限制错误，并提出是否转向C++以提升程序效率的问题。文章强调，在学习初期，掌握算法比追求语言效率更为重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目3：Perfect Subarray

官方都提示了本题如果用慢的语言过不了test 2，那要不要学C++呢，可我觉得目前还是应该着重学习算法，我还不到追求程序效率的时候。。。
看了前排大佬代码，改写成了Python，果不其然test 2 TLE了。。

T=int(input())
for tt in range(T):
    N=int(input())
    array=[int(s) for s in input().split()]
    min_presum=0
    dict1={}
    presum=0
    ans=0
    dict1[0]=1
    for i in range(len(array)):
        presum=presum+array[i]        
        T=presum
        j=0
        while T>=min_presum:
            T=presum-j**2
            j=j+1
            if T in dict1.keys():
                ans=ans+dict1[T]
        if presum in dict1.keys():
            dict1[presum]+=1
        else:
            dict1[presum]=1
        min_presum=min(min_presum,presum) # 更新最小的前缀和
    print("Case #{}: {}".format(tt+1, ans))