E - TT 的神秘任务3

题意:

TT 猫咖的生意越来越红火，人越来越多，也越来越拥挤。

为了解决这个问题，TT 决定扩大营业规模，但猫从哪里来呢？

TT 第一时间想到了神秘人，想要再次通过完成任务的方式获得猫咪。

而这一次，神秘人决定加大难度。

给定一个环，A[1], A[2], A[3], … , A[n]，其中 A[1] 的左边是 A[n]。要求从环上找出一段长度不超过 K 的连续序列，使其和最大。

这一次，TT 陷入了沉思，他需要你们的帮助。

Input
第一行一个整数 T，表示数据组数，不超过 100。

每组数据第一行给定两个整数 N K。（1 ≤ N ≤ 100000, 1 ≤ K ≤ N）

接下来一行，给出 N 个整数。（-1000 ≤ A[i] ≤ 1000）。

Output
对于每一组数据，输出满足条件的最大连续和以及起始位置和终止位置。

如果有多个结果，输出起始位置最小的，如果还是有多组结果，输出长度最短的。

Sample Input

4
6 3
6 -1 2 -6 5 -5
6 4
6 -1 2 -6 5 -5
6 3
-1 2 -6 5 -5 6
6 6
-1 -1 -1 -1 -1 -1

Sample Output

7 1 3
7 1 3
7 6 2
-1 1 1

思路:

首先是将环状拆成线性，在后面补K-1个数，是原数组的第1~(k-1)个数;
初始化每个点的前缀和
状态转移方程ans=max(ans,sum[i]-min{sum[m]})（i-k<=m<=i）
利用滑动窗口维护一个长度不超过k的sum单调递增序列，则min{sum[m]}就是队列的最左端
根据规则更新ans
如果有多个结果，输出起始位置最小的，如果还是有多组结果，输出长度最短的。

代码:

#include <algorithm>
#include <cstdio>
#include <deque>
using namespace std;
const int maxn = 2 * (100000 + 5);
int n, k, a[maxn], sum[maxn];
long long ans[3];
deque<int> q;

int main()
{
    int T;
    scanf("%d", &T);
    while (T--)
    {
        q.clear();
        ans[0] = -0x7fffff;
        scanf("%d %d", &n, &k);

        for (int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i]);
            sum[i] = sum[i - 1] + a[i];   //前缀和
        }

        for (int i = n + 1; i < n + k; i++)
            sum[i] = sum[i - 1] + a[i - n];      //链式前缀和

        for (int i = 1; i < n + k; i++)
        {//单调队列模板
            while (!q.empty() && sum[i - 1] < sum[q.back()])
                q.pop_back();    //维护窗口递增
            while (!q.empty() && q.front() < i - k)
                q.pop_front();     //维护窗口大小
            q.push_back(i - 1);   
            if (sum[i] - sum[q.front()] > ans[0])
            {
                ans[0] = sum[i] - sum[q.front()];
                ans[1] = q.front() + 1;     //窗口的左端点
                ans[2] = i;    //右端点
            }
        }

        if (ans[2] > n)
            ans[2] %= n;  //返回链状
        printf("%lld %lld %lld\n", ans[0], ans[1], ans[2]);
    }
    return 0;
}