JZOJ 1330. 迎接仪式

Description

　　LHX教主要来X市指导OI学习工作了。为了迎接教主，在一条道路旁，一群Orz教主er穿着文化衫站在道路两旁迎接教主，每件文化衫上都印着大字。一旁的Orzer依次摆出“欢迎欢迎欢迎欢迎……”的大字，但是领队突然发现，另一旁穿着“教”和“主”字文化衫的Orzer却不太和谐。
　　为了简单描述这个不和谐的队列，我们用“j”替代“教”，“z”替代“主”。而一个“j”与“z”组成的序列则可以描述当前的队列。为了让教主看得尽量舒服，你必须调整队列，使得“jz”子串尽量多。每次调整你可以交换任意位置上的两个人，也就是序列中任意位置上的两个字母。而因为教主马上就来了，时间仅够最多作K次调整（当然可以调整不满K次），所以这个问题交给了你。

Input

　　输入文件的第1行包含2个正整数N与K，表示了序列长度与最多交换次数。
　　第2行包含了一个长度为N的字符串，字符串仅由字母“j”与字母“z”组成，描述了这个序列。

Output

　　输出文件仅包括一个非负整数，为调整最多K次后最后最多能出现多少个“jz”子串。

Sample Input

5 2
zzzjj

Sample Output

2

Data Constraint

Hint

【样例说明】
　　第1次交换位置1上的z和位置4上的j，变为jzzzj；
　　第2次交换位置4上的z和位置5上的j，变为jzzjz。
　　最后的串有2个“jz”子串。

【数据规模】
　　对于10%的数据，有N≤10；
　　对于30%的数据，有K≤10；
　　对于40%的数据，有N≤50；
　　对于100%的数据，有N≤500，K≤100。

分析

首先相同字符是不用调换的，一个字符最多被调换一次(a<—>b,b<—>c等价于a<—>c)
f[i][j][k]表示前i个字符，改变了j个’j’和k个’z’后的“jz”串数。
那么只考虑前两位，有四种情况（jj,jz,zj,zz）来转移。
注意初始化！

代码

#include <algorithm>
#include <iostream>
#include <cstring>
#include <complex>
#include <cstdio>
#include <queue>
#include <cmath>
#include <map>
#include <set>

#define N 505
#define INF 0x7fffffff
#define sqr(x) ((x) * (x))
#define pi acos(-1)

int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9'){if (ch == '-') f = -1; ch = getchar();}
    while (ch >= '0' && ch <= '9')  {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

int f[N][105][105];
char s[N];

int main()
{
    int ans = 0;
    int n = read(), m = read();
    scanf("%s", s + 1);
    memset(f, -0x3f, sizeof(f));
    f[0][0][0] = 0; f[1][0][0] = 0;
    if (s[1] == 'z')
        f[1][0][1] = 0;
        else f[1][1][0] = 0;
    for (int i = 2; i <= n; i++)
        for (int j = 0; j <= m; j++)
            for (int k = 0; k <= m; k++)
            {
                f[i][j][k] = f[i - 1][j][k];
                if (s[i] == 'z' && s[i - 1] == 'j')
                    f[i][j][k] = std::max(f[i][j][k], f[i - 2][j][k] + 1);
                if (s[i] == 'z' && s[i - 1] == 'z' && k) 
                    f[i][j][k] = std::max(f[i][j][k], f[i - 2][j][k - 1] + 1);
                if (s[i] == 'j' && s[i - 1] == 'j' && j) 
                    f[i][j][k] = std::max(f[i][j][k], f[i - 2][j - 1][k] + 1);
                if (s[i] == 'j' && s[i - 1] == 'z' && j && k) 
                    f[i][j][k] = std::max(f[i][j][k], f[i - 2][j - 1][k - 1] + 1);
                if (j == k)
                    ans = std::max(ans, f[i][j][k]);
            }
    printf("%d\n",ans);
}