CF - 1000B. Light It Up

题意：在0时刻，灯的状态是打开，然后在每个a[i]处灯的开关状态反转一次，在m时刻灯的状态是关闭。要求是：在a数组中插入一个[0，m]中的数或者不插入，计算所有情况中最大的灯亮总时间

算法：前缀和 、思维

题解：在a[i]和a[i+1]之间插入一个数x时，a[i]前面的灯亮的时间和是不变的，x后面每个a数组的元素代表的状态反转一次，反转后a[i+1]之后灯亮的时间是固定的（不因为x的大小改变）。所以主要看a[i]和a[i+1]之间的状态。这时分为两种情况：

（1）a[i]代表开，a[i+1]代表关。则插入x后，a[i]到x代表灯亮，x到a[i+1]代表灯灭，为了使灯亮的时间最长，所以x应该是a[i+1]-1。则现在a[i+1]之前的灯亮的时间是之前的减1。

（2）a[i]代表关，a[i+1]代表开。则插入x后，a[i]到x代表灯灭，x到a[i+1]代表灯亮，为了使灯亮的时间最长，所以x应该是a[i]+1。则现在a[i+1]之前的灯亮的时间也是之前的减1。

得到这个结论之后，计算插入x后的灯亮总时间时，分为两部分；第一部分为0到a[i+1]，灯亮时间为sum[i]-1；第二部分为a[i+1]到m，灯亮时间为之前总的灯灭时间减去a[i+1]之前的灯灭时间。即((m-sum[n+1]) - (a[i]-sum[i]))。答案取max即可，记得和不插入数时的灯亮时间sum[n+1]作比较。

代码：时间复杂度O(n)

#include <algorithm>
#include <iostream>
#include <stdio.h>
#include <iterator>
#include <vector>
#include <cstring>
#include <string>
#include <cmath>
#include <deque>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
using namespace std;

typedef long long ll;
#define inf 0x3f3f3f3f
#define endl "\n"
#define IOS ios::sync_with_stdio(0);
#define debug(x) cout<<"**"<<x<<"**"<<endl;
const double eps = 1e-6; 
const double pi = acos(-1.0);
const int mod = 998244353;
const int N = 1e5 + 100;

int n, m, f;
int a[N], sum[N];

int main()
{
#ifndef ONLINE_JUDGE
	freopen("in.in", "r", stdin);
	freopen("out.out", "w", stdout);
#endif
    IOS;
	cin>>n>>m;
	for(int i = 1; i <= n; i++) cin>>a[i];
	a[n+1] = m;
	for(int i = 1; i <= n+1; i++)
	{
		f ^= 1;
		sum[i] = sum[i-1] + f*(a[i] - a[i-1]);
	}
	int ans = sum[n+1];
	for(int i = 1; i <= n; i++)
		ans = max(ans, (sum[i+1]-1) + ((m-sum[n+1]) - (a[i+1]-sum[i+1])));
    cout<<ans;
	return 0;
}

后记：如有疑问或者题解有错误，还请留言