AcWing 3810. 最长连续休息时间

原题链接

Description

一天可以被分为 $n$ 个时段。

一个工人的每日工作安排可以用一个长度为 $n$ 的 01 序列 $a_1,a_2,…,a_n$ 来表示。

$a_i$ 为 0 表示第 $i$ 个时间段是工作时间，$a_i$ 为 1 表示第 $i$ 个时间段是休息时间。

工人日复一日的严格按照这个工作安排来进行工作和休息。

请问，工人的最长连续休息时间有多长（单位：时段）？

注意，连续休息时间可能跨天。

保证工人至少在一个时间段处于工作状态。

Input

第一行包含整数 T，表示共有 T 组测试数据。

每组数据第一行包含整数 $n$。

第二行包含 $n$ 个整数$a_1,a_2,…,a_n$

Output

每组数据输出一行结果，表示最长连续休息时间。

Solution

连续休息时间可能会跨天，也就是第一天的末尾，和第二天的开头也可能会休息，所以，只要开辟两倍的数组，该问题转化为求 连续字符为1的最长子数组长度。

Code

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 400010;

int n;
int w[N];

int main()
{
    int T;
    scanf("%d", &T);
    while (T -- )
    {
        scanf("%d", &n);
        for (int i = 0; i < n; i ++ )
        {
            scanf("%d", &w[i]);
            w[n + i] = w[i];
        }

        int res = 0;
        for (int i = 0, s = 0; i < n * 2; i ++ )
            if (w[i])
            {
                s ++ ;
                res = max(res, s);
            }
            else s = 0;
        printf("%d\n", res);
    }
    return 0;
}