题意:
给出一个字符串,只包含7种字符,接下来给出一个数m,每次可以选择一段长度大于等于m的子串,并且子串中每个字母都相同,然后将其删除,删除后,两段自动补齐,问是否可以将其删完。
题解:
我们先预处理出字符串的每一段字母相同的连续区间,总共有sum个,然后只对1 到 sum 这段区间考虑。
将每种字母标号,设dp [i] [j] [k]表示 i 到 j 区间删完只剩标号为k的字母,且其最大的数量,f [i] [j] 表示 i 到 j 这段区间是否能删完。
对于 i 到 j 这段区间,假如第 i 个能单独删掉并且第 i+1 到 j 这段区间能删掉,那么 i 到 j 这段区间就能删掉,否则就算出 i+1 到 j 这段区间删掉只剩 i 这种字母的最大数量,即 dp [i+1] [j] [k], 过程中通过记忆化搜索进行优化。
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int MAXN=5e2+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
map<char ,int >mp;
pair<int,int>q[MAXN];
int dp[MAXN][MAXN][8];
char s[MAXN];
int f[MAXN][MAXN];
int m;
bool solve(int l,int r);
int dfs(int l,int r,int colar)
{
if(l>r) return 0;
if(dp[l][r][colar]!=-INF) return dp[l][r][colar];
int ans=-INF;
if(solve(l,r)) ans=0;
for(int i=l;i<=r;i++)
{
if(q[i].first==colar&&solve(l,i-1))
{
ans=max(ans,q[i].second+dfs(i+1,r,colar));
}
}
return dp[l][r][colar]=ans;
}
bool solve(int l,int r)
{
if(l>r) return 1;
if(f[l][r]!=-1) return f[l][r];
int col=q[l].first;int num=q[l].second;
if(num>=m&&solve(l+1,r))
{
return f[l][r]=1;
}
return f[l][r]=(num+dfs(l+1,r,col)>=m);
}
int main()
{
memset(f,-1,sizeof f);
cin>>s+1;
cin>>m;
int n=strlen(s+1);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
for(int k=1;k<=7;k++)
{
dp[i][j][k]=-INF;
}
}
}
mp['R']=1;mp['G']=2;mp['B']=3;
mp['C']=4;mp['M']=5;mp['Y']=6;
mp['K']=7;
int sz=0;
int cnt=1;
for(int i=2;i<=n+1;i++)
{
if(s[i]==s[i-1])
{
cnt++;
}
else
{
q[++sz].first=mp[s[i-1]];
q[sz].second=cnt;
cnt=1;
}
}
if(solve(1,sz))
{
cout<<"Yes"<<endl;
}
else
{
cout<<"No"<<endl;
}
}