【AtCoder - abc310_e】NAND repeatedly【找规律】

分析

看完题面终于知道NAND是什么意思（not and），想到这个应该是线性的时间复杂度。

进一步看，这个运算有个规律，对于任意一个区间，只要结尾是0，那运算结果就是1，对于连续的1，结果0,1交替出现。

考虑对于每个位置为结尾的区间统计答案，先把所有单点为1的加进答案，然后找到所有结尾为0的位置 $i$ ，然后答案加上 $i-1$ 。对于为1的区间，发现答案交替的顺序与结尾连续1的个数有关，这个预处理是很容易的，以 $i$ 结尾的区间连续1的个数记为 $s[i]$，如果区间结尾是奇数个1，那么只有从最后一个0开始的区间有贡献，再加上 $s[i]/2$ 就行（交替1）；如果偶数个1结尾，前面 $i-s[i]-1$ 个都有贡献，再加上 $s[i]/2-1$ 就行（单点的最后那个统计过了）

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;

ll n,a[1000010],s[1000010];
ll ans;

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		char c;
		cin>>c;
		a[i]=c-48;
		ans+=a[i];
	}
	int cnt=0;
	for(int i=1;i<=n;i++)
	{
		if(a[i]==1) cnt++;
		else cnt=0,ans+=i-1;
		s[i]=cnt;
	}
	for(int i=1;i<=n;i++)
	{
		if(!a[i]) continue;
		//s-1个区间全部都是1，特判奇偶
		if(s[i]%2==1)
		{
			if(i-s[i]>=1) ans++;
			ans+=s[i]/2;
		}
		else
		{
			if(i-s[i]>=1) ans+=i-s[i]-1;
			ans+=s[i]/2-1;
		}
	}
	cout<<ans;
	return 0;
}