题意:n头牛站成一条线,有朝前有朝后的的。每次可以且只能将连续k个牛全部转向,问能够使得所有牛全部面朝前方的最大的k。确定k后,求出最小的翻转次数。
思路:首先可以肯定,一旦确定k,翻转次数是确定的,从左到右扫一遍即可。接下来想k怎么确定,没什么好办法只能从大到小枚举(不满足单调性,不能用二分)。下面再考虑怎么扫,如果直接暴力,复杂度是Σk(n-k)(k从1到n),经过计算这是n^3的,不可接受。
优化的方法是计算第i头牛是否要翻转的时候,只需要知道第i-k+1头到第i头之间的翻转次数,那么维护这个次数即可。每次向右移动一格,需要看看左边出去的那格(第i-k格)是翻转了没有,维护好即可。这样扫一遍的复杂度是O(n)。算法的总共时间复杂度是O(n^2)。nice~
#include <cstdio>
#include <queue>
#include <algorithm>
#include <cstring>
using namespace std;
#define INF 0x3fffffff
#define clr(s,t) memset(s,t,sizeof(s));
#define N 5005
int s[N],sum[N],now[N],n;
int main(){
int i,j,k,res;
char ch;
scanf("%d\n",&n);
for(i = 1;i<=n;i++){
ch = getchar();
if(ch == 'F')
s[i] = 1;
else
s[i] = 0;
getchar();
}
sum[0] = 0;
for(k = n;k>=1;k--){
res = 0;
for(i = 1;i<=n-k+1;i++){
if(i>k)
sum[i-1] -= now[i-k];
if((sum[i-1]+s[i])&1){
sum[i] = sum[i-1];
now[i] = 0;
}else{
sum[i] = sum[i-1]+1;
now[i] = 1;
res++;
}
}
for(;i<=n;i++){
if(i>k)
sum[i-1] -= now[i-k];
if(!((sum[i-1]+s[i])&1))
break;
sum[i] = sum[i-1];
}
if(i>n)
break;
}
printf("%d %d\n",k,res);
return 0;
}