The 2020 ICPC Asia Taipei-Hsinchu Site Programming Contest E - A Color Game

题意:
给出一个字符串,只包含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
#include
#include
#include
#include
#include
#include
#include
#include
#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;
    }
}

你可能感兴趣的:(The 2020 ICPC Asia Taipei-Hsinchu Site Programming Contest E - A Color Game)