USACO 喂饱奶牛

题目背景

时/空限制：1s / 64MB

题目描述

Farmer John 有 N 头奶牛，每头奶牛的品种是更赛牛（Guernsey）或荷斯坦牛（Holstein）之一。

她们沿水平方向排成一行，奶牛们占据的位置编号为 1…N 。

由于奶牛们都饿了，FJ 决定在 1…N 中的某些位置上种植草地。

更赛牛和荷斯坦牛喜欢不同类型的草，所以如果 Farmer John 决定在某个位置种草，他必须选择种植更赛牛喜欢的草或荷斯坦牛喜欢的草——他不能在同一个位置同时种两种草。

种植的每一片草地都可以喂饱数量不限的相应品种的奶牛。

每头奶牛愿意移动至多 K （0≤K≤N−1 ）个位置以前往一个草地。

求出喂饱所有奶牛所需种植的最小草地数量。

此外，输出一种使用最小草地数量喂饱所有奶牛的种植方案。

任何满足上述条件的方案均视为正确。

输入格式

每个测试用例包含 T 个子测试用例，每个用来描述一种奶牛的排列。

输入的第一行包含 T 。以下是 T 个子测试用例。

每个子测试用例的第一行包含 N 和 K 。

第二行包含一个长度为 N 的字符串，其中第 i 个字符表示第 i 头奶牛的品种（G 表示更赛牛，H 表示荷斯坦牛）。

输出格式

对于 T 个子测试用例中的每一个，输出两行。

第一行输出喂饱所有奶牛所需的最小草地数量。

第二行输出一个长度为 N 的字符串，表示一种使用最小草地数量喂饱所有奶牛的种植方案。第 i 个字符表示第 i 个位置，若不种草则为 .，若种植喂饱更赛牛的草则为 G，若种植喂饱荷斯坦牛的草则为 H。

任何合法的方案均可通过。

输入输出样例

输入 #1复制

6
5 0
GHHGG
5 1
GHHGG
5 2
GHHGG
5 3
GHHGG
5 4
GHHGG
2 1
GH

输出 #1复制

5
GHHGG
3
.GH.G
2
..GH.
2
...GH
2
...HG
2
HG

说明/提示

数据范围
1≤T≤10 ,
1≤N≤10^5 ,
0≤K≤N−1
样例解释
注意对于某些子测试用例，存在多种可通过的方案使用最小数量的草地。例如，在第四个子测试用例中，以下是另一个可以通过的答案：

.GH..

这个方案在第二个位置种植一块喂饱更赛牛的草地以及在第三个位置种植一块喂饱荷斯坦牛的草地。这使用了最小数量的草地并确保了所有奶牛都在她们喜欢的草地的 3 个位置以内。

AC代码：

#include <bits/stdc++.h>
using namespace std;
int t, n, k;
char cow[100005] = {0}, ans[100005];
int main() {
	cin >> t;
	for (int i = 1; i <= t; i++) {
		int cnt = 0;
		memset(ans, '.', sizeof(ans));
		cin >> n >> k;
		for (int j = 1; j <= n; j++) {
			cin >> cow[j];
		}
		int mark = 1;
		while (mark <= n) {
			if (cow[mark] == 'G') {
				if ((mark + k) < n) {
					ans[mark + k] = 'G';
					mark = mark + 2 * k + 1;
				} else {
					ans[n] = 'G';
					break;
				}
			} else mark++;
		}
		int mark2 = 1;
		while (mark2 <= n) {
			if (cow[mark2] == 'H') {
				if ((mark2 + k) < n) {
					ans[mark2 + k] = 'H';
					mark2 = mark2 + 2 * k + 1;
				} else {
					if (ans[n] == 'G') {
						ans[n - 1] = 'H';
						break;
					} else {
						ans[n] = 'H';
						break;
					}
				}
			} else mark2++;
		}
		for (int i = 1; i <= n; i++) {
			if (ans[i] != '.') cnt++;
		}
		cout << cnt << endl;
		for (int i = 1; i <= n; i++) {
			cout << ans[i];
		}
		cout << endl;
	}
	return 0;
}