HDU 5583 Kingdom of Black and White(模拟)

题意：给出一个01串，他的价值就是每段连续的0或者1的个数的平方。现在至多一次机会变换一个位置的0或者1，问能得到最大的价值是多少。

解法：很明显，让大的尽量大，这是一种贪心思想，但是仔细分析发现又不是绝对的，要看答案的贡献度，所以我们采取暴露枚举的方法即可。

首先预处理出每个串的长度。

然后暴力，使得左边-1自己+1，或者右边-1自己+1，看是否可以更新最大值。（注意边界）

最后输出答案即可。

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5;
long long num[maxn];
char str[maxn];

long long Count(int x, int y) {
return (num[x] + 1) * (num[x] + 1) + (num[y] - 1) * (num[y] - 1) - (num[x] * num[x] + num[y] * num[y]);
}

int main() {
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
#endif
int T, Case = 1;
scanf("%d",&T);
while(T--) {
memset(num, 0, sizeof(num));
scanf("%s", str);
int len = strlen(str);
int idx = 1;
for(int i = 0, j; i < len; i++) {
long long cnt = 1;
for(j = i + 1; j < len; j++) {
if(str[j] != str[i]) {
break;
}
cnt++;
}
num[idx++] = cnt;
i = j - 1;
}
long long sum = 0;
for(int i = 0; i < idx; i++) {
sum += num[i] * num[i];
}
long long tmp, Max = 0;
for(int i = 1; i < idx; i++) {
if(num[i] >= num[i - 1] && i - 1 > 0) {
if(num[i - 1] == 1) {
tmp = (num[i] + 1 + num[i - 2]) * (num[i] + 1 + num[i - 2]);
tmp -= (num[i] * num[i] + num[i - 2] * num[i - 2] + 1);
} else {
tmp = Count(i, i - 1);
}
if(tmp > Max)
Max = tmp;
}
if(num[i] >= num[i + 1] && i + 1 < idx) {
if(num[i + 1] == 1) {
tmp = (num[i] + 1 + num[i + 2]) * (num[i] + 1 + num[i + 2]);
tmp -= (num[i] * num[i] + num[i + 2] * num[i + 2] + 1);
} else {
tmp = Count(i, i + 1);
}
if(tmp > Max)
Max = tmp;
}
}
printf("Case #%d: %lld\n",Case++, sum + Max);
}
return 0;
}