P2882 [USACO07MAR]Face The Right Way（差分枚举）

题目
思路：
记F为1，B为0
枚举k，遇到0就反转k个单位，但暴力复杂度为n^3会T
用差分复杂度n^2

#include<bits/stdc++.h>
using namespace std;
const int N=5010;
int n,kk,m;
bool a[N],B[N];
char c;
int main()
{
	m=N;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>c;
		if(c=='F')a[i]=1;
	}
	for(int k=1;k<=n;k++)
	{
		int cnt=0;
		bool f=1,b=0;
		memset(B,0,sizeof B);//B数组记录反转次数，反转2次等于不反转 ，所以B用bool类型开 
		for(int i=1;i<=n;i++)
		{
			b^=B[i];// b记录当前位置的翻转次数
			if(!(a[i]^b))//比如a[i]为1，b也为1，那么i这个位置为0，如果这个位置为0那么反转k个长度 
			{
				if(i+k-1>n){f=0;break;}//如果最后剩下的不满k个那么这个k不行，跳出
				b^=1,B[i+k]^=1;//差分 
				cnt++;
			}
		}
		if(f&&m>cnt)
		{
			kk=k;
			m=cnt;
		}
	}
	cout<<kk<<' '<<m<<'\n';
	return 0;
}