Problem Description
N头牛排成了一列。每头牛或者向前或者向后。为了让所有的牛都面向前方,农夫约翰买了一台自动转向的机器。这个机器在购买时就必须设定一个数值K,机器每操作一次恰好使K头连续的牛转向。请求出为了让所有牛都能面向前方需要的最少的操作次数M和对应的最小的K。
思路:
首先交换区间反转的顺序对结果是没有影响的,此外,可以知道对同一个区间进行两次以上的反转是多余的。
我们枚举K,然后从左往右,遇到需要反转的就给它反转,然后标记一下区间【i, i+k-1】被反转过。核心点就是在于如何标记能实现O(1)的复杂度。
用一个f[i]数组,f[i]: = 区间[i, i+k-1]进行了反转的话则为1,否则为0。这样,在考虑第i头牛时,如果区间[i-k+1,i-1]f[i]的和为奇数的话,则这头牛的方向与起始方向是相反的
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 5005;
int dir[N];//0面向前方,1面向后方
int f[N], n;
int cal(int k)
{
memset(f, 0, sizeof(f));
int sum = 0, res = 0;
for(int i = 0; i + k - 1 < n; i++)
{
if(i-k >= 0) {//[i-k+1,i-1]的和影响,所以减去f[i-k]
sum -= f[i-k];
}
if((dir[i]+sum)%2 != 0) {//sum偶数i这个位置不变 sum奇数i这个位置相反
res++;
f[i] = 1;
}
sum += f[i];
}
for(int i = n-k+1; i < n; i++)//检查剩下的牛是否有面朝后方
{
if(i-k >= 0) {
sum -= f[i-k];
}
if((dir[i]+sum)%2 != 0) {
return -1;
}
}
return res;
}
void solve()
{
int K, M = inf;
for(int k = 1; k <= n; k++)//枚举一次可以反转的区间,题目要求从1开始。
{
int m = cal(k);
if(m != -1 && M > m)
{
K = k, M = m;
}
}
printf("%d %d\n", K, M);
}
int main()
{
char s[2];
while(~scanf("%d", &n))
{
for(int i = 0; i < n; i++) {
scanf("%s", s);
if(s[0] == 'F') dir[i] = 0;
else dir[i] = 1;
}
solve();
}
return 0;
}