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;
}

猜你喜欢

转载自blog.csdn.net/weixin_46001550/article/details/108735517