题目链接:https://www.luogu.com.cn/problem/P2882
洛谷P2882 [USACO07MAR]Face The Right Way G
暴力,差分
题目大意为给定n个字符表示n头牛的朝向,每次可以选择连续的k头牛,将其方向反转,求最小翻转次数与对应的最小k,使得所有牛方向朝前。
用1表示朝前的状态,0表示朝后的状态,用一个长度为n的数列表示n头牛的状态,则题意为每次可以将长度为k的区间内的所有数字反转,要求将所有数字变为1。从1开始往后遍历,若第i个数之前的所有数都为1,则反转起点一定在当前位置之后,因为反转前面的数字没有意义。所以从小到大枚举所有的k,对于每个k,从数列左端开始向右遍历,遇到当前位置数字为0,则反转之后的k个数,这个地方的反转操作不用一一遍历,可以用一个差分数组f存储每个位置被反转的次数,最后判断每个位置是否为1只需判断原数列a的每一项与差分数列f求前缀和后的每一项之和ai+fi的奇偶性即可。最后存储枚举过程中的最小操作次数即可。
#include<bits/stdc++.h>
#define ll long long
#define next next_
#define y1 yy
using namespace std;
int n,sum,ans=1e9+7,K,a[5010],f[5010],cnt;
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
char c;
cin>>c;
a[i]=c=='F';
cnt+=a[i];
}
for(int k=1;k<=n;k++){
memset(f,0,sizeof(f));
sum=0;
for(int i=1;i<=n-k+1;i++){
f[i]+=f[i-1];
if((a[i]+f[i])%2==0){
f[i]++;
f[i+k]--;
sum++;
}
}
bool ok=true;
for(int i=n-k+2;i<=n;i++){
f[i]+=f[i-1];
if((a[i]+f[i])%2==0){
ok=false;
break;
}
}
if(ok){
if(sum<ans){
ans=sum;
K=k;
}
}
}
printf("%d %d",K,ans);
return 0;
}