美团回转寿司 Python （未通过，但是也不知道哪错了）

最新推荐文章于 2022-04-19 10:04:36 发布

原创最新推荐文章于 2022-04-19 10:04:36 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

牛客网剑指Offer 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个寻找连续寿司和最大值的问题，给出了一段Python代码实现。该算法利用前缀和找到最优解，对于输入的测试用例能够正确计算。通过遍历前缀和数组并维护最小前缀和及其索引，找出最大连续子数组和。

问题

3.24：不知道自己哪里写错了，只是通过了 3/10。还很自信的觉得自己没写错 >_<
麻烦走过路过的大佬帮忙瞅下，看看还有啥其他的测试用例，边界条件是我没想到的

题目

小美请小团吃回转寿司。转盘上有N盘寿司围成一圈，第1盘与第2盘相邻，第2盘与第3盘相邻，…，第N-1盘与第N盘相邻，第N盘与第1盘相邻。小团认为第i盘寿司的美味值为A[i]（可能是负值，如果小团讨厌这盘寿司）。现在，小团要在转盘上选出连续的若干盘寿司，使得这些寿司的美味值之和最大（允许不选任何寿司，此时美味值总和为0）。
输入描述

第一行输入一个整数T（1<=T<=10），表示数据组数。
每组数据占两行，第一行输入一个整数N（1<=N<=10^5）；
第二行输入N个由空格隔开的整数，表示A[1]到A[N]（-10^4<=A[i]<=10^4）。

输出描述

每组数据输出占一行，输出一个整数，表示连续若干盘寿司的美味值之和的最大值。

输入测试用例

1
4
3 -2 4 -1

输出测试用例

6

Python 代码

import sys
if __name__ == '__main__':
    T = int(input())
    for _ in range(T):
        N = int(input())
        A = list(map(int, input().split()))
        prefix = [0]
        prefix_sum = 0
        A.extend(A)
        for i in range(2*N):
            prefix_sum += A[i]
            prefix.append(prefix_sum)
        # print(prefix)
        min_prefix, min_idx, ans = sys.maxsize, 0, 0
        for i in range(1, len(prefix)):
            if i - min_idx > N:
                # print(i, " min_idx: ", min_idx)
                break
            ans = max(ans, prefix[i] - min_prefix)
            if prefix[i] <= min_prefix:
                min_prefix = prefix[i]
                min_idx = i
        print(ans)