洛谷 P1136 迎接仪式

题目描述

LHX教主要来X市指导OI学习工作了。为了迎接教主，在一条道路旁，一群Orz教主er穿着文化衫站在道路两旁迎接教主，每件文化衫上都印着大字。一旁的Orzer依次摆出“欢迎欢迎欢迎欢迎……”的大字，但是领队突然发现，另一旁穿着“教”和“主”字文化衫的Orzer却不太和谐。

为了简单描述这个不和谐的队列，我们用“j”替代“教”，“z”替代“主”。而一个“j”与“z”组成的序列则可以描述当前的队列。为了让教主看得尽量舒服，你必须调整队列，使得“jz”子串尽量多。每次调整你可以交换任意位置上的两个人，也就是序列中任意位置上的两个字母。而因为教主马上就来了，时间仅够最多作K次调整（当然可以调整不满K次），所以这个问题交给了你。

输入输出格式

输入格式：

输入文件welcome.in的第1行包含2个正整数N与K，表示了序列长度与最多交换次数。

第2行包含了一个长度为N的字符串，字符串仅由字母“j”与字母“z”组成，描述了这个序列。

输出格式：

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

输入输出样例

输入样例#1：

5 2 
zzzjj

输出样例#1：

2

说明

【样例说明】

第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。

f[i][j][g]表示前i个字符，换了j个'j'，换了g个'z'。

所以答案就是f[n][i][i]。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,k,ans,f[505][105][105];
char s[505];
int main()
{
	scanf("%d%d%s",&n,&k,s+1);
	memset(f,0xaf,sizeof(f));
	f[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=1;i<=n;i++)
		for(int j=0;j<=k;j++)
			for(int g=0;g<=k;g++)
			{
				f[i][j][g]=f[i-1][j][g];
				if(s[i-1]=='j'&&s[i]=='j'&&j)
					f[i][j][g]=max(f[i][j][g],f[i-2][j-1][g]+1);
				if(s[i-1]=='j'&&s[i]=='z')
					f[i][j][g]=max(f[i][j][g],f[i-2][j][g]+1);
				if(s[i-1]=='z'&&s[i]=='z'&&g)
					f[i][j][g]=max(f[i][j][g],f[i-2][j][g-1]+1);
				if(s[i-1]=='z'&&s[i]=='j'&&j&&g)
					f[i][j][g]=max(f[i][j][g],f[i-2][j-1][g-1]+1);
			}
	for(int i=0;i<=k;i++)
		ans=max(ans,f[n][i][i]);
	printf("%d\n",ans);
	return 0;
}