POJ 3276 反转(开关问题)

点击打开链接

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int M =5100;
int dir[M];//  dir[i] 0:F 1;B
int n; 
int f[M]; // f[i] i~i+k-1的区间是否被翻转过 
 
int calc(int k) //判断第i只cow的State 只要判断 j=i-k+1~i-1 f[j]的累加和 %2 为0则State和初始State相同  
{
	memset(f,0,sizeof(f));
	int res=0;
	int sum=0;
	for(int i=0;i+k-1<=n-1;i++)
	{
		if((dir[i]+sum)%2!=0)
		{
			res++;
			f[i]=1;
		}
		
		sum+=f[i];
		if(i-k+1>=0) // 第i+1cow的State = j=i-k+1~i-1 f[j] + f[i] -f[i-k+1]     //j=i-k+1~i-1多出f[i-k+1] 
		{
			sum-=f[i-k+1];	
		} 
		 
		
	}
	for(int i=n-k+1;i<n;i++)// Check之后的cow的State 
	{
		if((dir[i]+sum)%2!=0) //无解 
		{
			return -1;
		}
		
		if(i-k+1>=0)
		sum-=f[i-k+1];
	}
	return res;
}



			//最左端的cow区间翻转后 之后的区间时候翻转就固定下来了 
void solve() //包含最左端的cow的区间只有一个  如果是 F 则f[i]开头的区间就不必要翻转 ,找到第一个为B的COW进行翻转 反转后继续找到"最左端为B的cow进行反转即可"  
{
	int K=1,M=n;
	for(int k=1;k<=n;k++) //一次翻转k格 
	{
		int m=calc(k); //最小步数m 

		if(m>0&&m<M)
		{
			M=m;
			K=k; 
		}
		
	} 
	 cout<<K<<" "<<M; 
	
}
int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
	{
		char c;
		cin>>c;
		if(c=='B')
		{
			dir[i]=1;	
		} 
		else
		{
			dir[i]=0;
		}
	}
	solve();
	return 0;	
}