AcWing 3774. 亮灯时长(贪心+枚举+前缀和)

思路好难想
参考讲解视频:https://www.acwing.com/video/3318/

题目

自习室内有一个智能灯。
在 0 时刻,管理员会将打开电闸,并将灯点亮。
在 M 时刻,管理员会直接拉下电闸,此时,如果灯处于点亮状态,则会因为断电而熄灭。
在 0∼M 之间有 n 个不同时刻,不妨用 a1,a2,…,an 表示,其中 0<a1<a2<…<an<M。
在这 n 个时刻中的每个时刻,管理员都会拨动一次智能灯的开关,使灯的状态切换(亮变灭、灭变亮)。
现在,你可以最多额外指定一个时刻(也可以不指定),让管理员在此时刻也拨动开关一次。注意选定的时刻不能与 a1,a2,…,an 相等。
你的目的是让亮灯的总时长尽可能长。
输出这个最大亮灯总时长。

输入输出

输入:第一行包含整数 T,表示共有 T 组测试数据。
每组数据,第一行包含两个整数 n 和 M。
第二行包含 n 个整数 a1,a2,…,an。
输出:输出一个整数,表示最大亮灯总时长。

思路

有n个时间点,将0-M分成n+1个段。画图理解更直观。
分情况讨论:
指定按下开关的时刻处于奇数段a[i-1]-a[i]:当前时间段长度为t,亮灯时长总和为i-1前奇数段之和 + i段后偶数段之和 + (t-1),(t-1)是在a[i-1]-a[i]段获得的亮灯时长最大值。
指定按下开关的时刻处于偶数段a[i-1]-a[i]:与奇数段一样的方法,可以得到和奇数段一样的结果。
前缀和
计算i-1前奇数段之和、i段后偶数段之和需要用到前缀和,判断是奇数段还是偶数段,加到两个前缀和数组中。
用贪心计算最大的段来获得最大的t-1

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

using namespace std;

const int N = 100010;
typedef long long LL;
LL a[N],s[N],t[N];  //s[i]是第i个时间点之前奇数段之和,t[i]是偶数段
int main()
{
    int T;
    cin >> T;
    while (T --){
        int n,m;
        cin >> n >> m;
        LL res = 0;
        for (int i = 1; i <= n; i ++ ) cin >> a[i];
        a[n+1] = m;     //把m算作最后一个时间点
        for (int i = 1; i <= n+1; i ++ ){
            s[i] = s[i-1];
            t[i] = t[i-1];
            if(i%2) s[i] += a[i] - a[i-1];  //计算奇数段前缀和
            else t[i] += a[i] -a[i-1];  //计算偶数段前缀和
        }
        res = s[n+1];   //res初始化为不需要指定时刻开关时的答案,即所有奇数段之和
        for (int i = 1; i <= n+1; i ++ ){
            int l = a[i] - a[i-1];  //l为每一段长度
            if(l == 1) continue;
            res = max(res,s[i-1]+t[n+1]-t[i]+l-1);  //更新最大亮灯时长
        }
        cout << res << endl;
    }
}
posted @ 2021-07-20 23:49  inss!w!  阅读(33)  评论(0编辑  收藏  举报