Face The Right Way POJ - 3276 枚举 差分

题解

题目大意，一排牛，有正朝农夫的有背朝农夫的，选取一个k，农夫可以将一段长度为k的牛进行翻转，问最少翻转次数和对应的k。

N不是很大，枚举k的值，检测当前的k是否能完成反转全部的牛并且得到最小的反转次数。
对于一个序列如FFBBFBBF无论k为何值，第一次反转的起点都要从一个B的位置来，否则会造成更多的B。
反转后又回到了一些F一些B的状态，依然从第一个B的位置反转，所以反转方式已经固定了。
但是如果直接循环对长度为k的序列进行反转则会超时，使用差分，反转区间[l, r]则t[l]++, t[r + 1]–，过程中求前缀和。
如果当前位置的值为奇数则表示为B为偶数表示F，初始化时也要用差分。这样反转就可以降到O(1)，总复杂度O(N^2)。

AC代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#define fst first
#define sed second
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const int N = 5e3 + 10;
int s[N], t[N];

int main()
{
#ifdef LOCAL
	freopen("C:/input.txt", "r", stdin);
#endif
	int n, m = INF, p = 1;
	cin >> n; 
	for (int i = 1; i <= n; ++i)
	{
		char c;
		scanf(" %c", &c);
		if (c == 'B')
			++s[i], --s[i + 1]; //差分 偶数表示F奇数表示B
	}
	for (int k = 1; k <= n; ++k) //枚举k
	{
		memcpy(t, s, sizeof(s));
		int cnt = 0; //操作次数
		for (int i = 1; i + k - 1 <= n; ++i) //整k
		{
			t[i] += t[i - 1]; //差分前缀和 
			if (t[i] & 1) //从有1的位置开始反转 不会从0开始
				++t[i], --t[i + k], ++cnt; //差分
		}
		int flag = 1;
		for (int i = n; i > n - k; --i) //检测是否有残留
			if (t[i] & 1)
				flag = 0;
		if (flag && cnt < m)
			m = cnt, p = k;
	}
	cout << p << " " << m << endl;

	return 0;
}