这是一道反转开关问题,想让所有的牛都面朝前方,令F为0, B为1,给定一个k值,可以从最左端的牛开始,如果这头牛面朝后方,那么以这头牛为开始往后的k头牛的区间都应该反转,再以下一头牛为起点,以此类推…
用f[i]记录以i为起点的牛是否反转,反转为1,否则为0.
同时用sum记录第i头牛的反转情况,如果sum为奇数则与原方向相反,用sum与原方向相加即可得出是否需要反转。
当所有区间都反转完毕后需要检验剩下的k-1头牛是否面向前方。
#include <cstdio>
#include <cstring>
const int maxn = 5000 + 10;
int dir[maxn];
int f[maxn];
int calc(int K, int n) {
memset(f, 0, sizeof(f));
int res = 0;
int sum = 0;
for (int i = 0; i + K <= n; i++) {
if ((dir[i] + sum) % 2 != 0) {
res++;
f[i] = 1;
}
sum += f[i];
if (i - K + 1 >= 0) {
sum -= f[i-K+1];
}
}
for (int i = n - K + 1; i < n; i++) {
if ((dir[i] + sum) % 2 != 0) {
return -1;
}
if (i - K + 1 >= 0) {
sum -= f[i-K+1];
}
}
return res;
}
int main(int argc, char const *argv[]) {
int n;
scanf("%d", &n);
getchar();
for (int i = 0; i < n; i++) {
char ch;
scanf("%c", &ch);
getchar();
if (ch == 'F') {
dir[i] = 0;
} else {
dir[i] = 1;
}
}
int K = 1, M = n;
for (int k = 1; k <= n; k++) {
int m = calc(k, n);
if (m >= 0 && M > m) {
M = m;
K = k;
}
}
printf("%d %d\n", K, M);
return 0;
}