PAT 1044 Shopping in Mars (25) python 尺取法

尺取法相关介绍参见https://blog.csdn.net/lxt_lucia/article/details/81091597。

本题可以抽象为给定一个正整数序列D，求其最短连续子序列，使其和大于等于M。

和连续子序列有关的问题一般都可以先进行预处理，得到D的前缀和序列S（S第一个元素为0）。位置 i 到 j 的区间和可以通过S[j] - S[i-1]得到。

由于本题要求推导出满足条件且和最小的所有的最短子序列，很多人（包括我）都采用了从头开始遍历i，通过二分法求出j的位置的办法。不过这个方法在C++下可行，Python3下就总有一个测试点过不去。在网上参考了大佬的博客，才得知有尺取法这个方法。

另一道题PAT 1085 Perfect Sequence 在 Python 3下使用二分仍然会有一个数据点超时，类似的优化方法可以参考这位大佬的博客：https://blog.csdn.net/qq_40438165/article/details/89741111

下面贴上Python 3 AC代码。

n,m = [int(x) for x in input().split() ]
d = [int(x) for x in input().split() ]

s = [0];
total = 0;
for dia in d:
    total += dia;
    s.append(total);

mini = 10 ** 8;
pairs = [];

j = 0;
for i in range(1,n+1):

    #注意下面的循环会使j沿一个方向一直移动到n，所有i遍历完成后总的时间复杂度为O(n)
    while j <= n and s[j] - s[i-1] < m:
        j += 1;

    if j > n:
        break;
    if s[j] - s[i-1] < mini:
        pairs.clear();
        pairs.append((i,j));
        mini = s[j] - s[i-1]
    elif s[j] - s[i-1] == mini:
        pairs.append((i,j));
    

for i,j in pairs:
    print("%d-%d" % (i,j));