AcWing 319. 折叠序列(区间dp)

题目传送门

题意: 给你一个长度不超过100的字符串,我们可以将它折叠,折叠规则如下: 如果有一个串多次重复,比如 a b a b a b ababab ababab,就可以改成 3 ( a b ) 3(ab) 3(ab) a a a a a a aaaaaa aaaaaa可以改成 6 ( a ) 6(a) 6(a), a b c c c a b c c c abcccabccc abcccabccc可以改成 2 ( a b c c c ) 2(abccc) 2(abccc)或者 2 ( a b 3 ( c ) ) 2(ab3(c)) 2(ab3(c))。问你对于给定字符串能折叠成的最小长度的字符串。

思路: 求最短的长度的字符串,我们可以先求出它的长度,详见这题。我们这里主要解决输出问题。
我们既然知道了我们可以折叠的最短长度,那么我们对 [ 1 , n ] [1,n] [1,n]去深度搜索。

如果
f [ i ] [ j ] = j − i + 1 f[i][j]=j-i+1 f[i][j]=ji+1,即这段里面没有折叠或者折叠之后长度不减少,我们就可以直接输出这一段。
否则
我们就还原折叠的过程。从l到r枚举一个k,如果发现 f [ i ] [ j ] = f [ i ] [ k ] + f [ k + 1 ] [ j ] f[i][j]=f[i][k]+f[k+1][j] f[i][j]=f[i][k]+f[k+1][j],那么 f [ i ] [ j ] f[i][j] f[i][j]就是 f [ i ] [ k ] f[i][k] f[i][k] f [ k + 1 ] [ j ] f[k+1][j] f[k+1][j]拼接起来的,那么直接去搜索 [ i , k ] [i,k] [i,k], [ k + 1 , j ] [k+1,j] [k+1,j]
还有就是
对于折叠起来的问题,我们也需要去判断x个 s [ i . . . k ] s[i...k] s[i...k]是否可以代替 s [ k + 1... j ] s[k+1...j] s[k+1...j]

代码:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define pii pair<int,int>
#define ull unsigned long long
#define pdd pair<double,double>
#define lowbit(x) x&-x
#define all(x) (x).begin(),(x).end()
#define sz(x) (int)(x).size()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read()
{
    
    
    int x=0,f=1;
    char ch=gc();
    while(ch<'0'||ch>'9')
    {
    
    
        if(ch=='-')
            f=-1;
        ch=gc();
    }
    while(ch>='0'&&ch<='9')
    {
    
    
        x=x*10+ch-'0';
        ch=gc();
    }
    return x*f;
}
using namespace std;
const int N=2e4+55;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-6;
const double PI=acos(-1);
char s[N];
int f[222][222],b[222];
bool canfold(int i,int j,int k)
{
    
    
    int len=j-i+1,p=k-i+1;
    if(len%p)
        return 0;
    int l=i,r=k+1;
    while(r<=j)
    {
    
    
        if(s[l]==s[r])
        {
    
    
            l++;r++;
        }
        else return 0;
    }
    return 1;
}
void dfs(int l,int r)
{
    
    
    if(f[l][r]==r-l+1)
    {
    
    
        for(int i=l;i<=r;i++)
            cout<<s[i];
        return ;
    }
    for(int k=l;k<r;k++)
    {
    
    
        if(f[l][r]==f[l][k]+f[k+1][r])
        {
    
    
            dfs(l,k);dfs(k+1,r);
            return ;
        }
    }
    for(int k=l;k<=r;k++)
    {
    
    
        if(canfold(l,r,k))
        {
    
    
            cout<<(r-l+1)/(k-l+1);
            cout<<'(';
            dfs(l,k);
            cout<<')';
            return ;
        }
    }
}
void solve()
{
    
    
    cin>>s+1;
    int n=strlen(s+1);
    for(int i=1;i<10;i++) b[i]=1;
    for(int i=10;i<100;i++) b[i]=2;
    b[100]=3;
    for(int i=1;i<=n;i++)
        for(int j=i;j<=n;j++)
            f[i][j]=j-i+1;
    for(int len=2;len<=n;len++)
    {
    
    
        for(int i=1;i+len-1<=n;i++)
        {
    
    
            int j=i+len-1;
            for(int k=i;k<j;k++)
            {
    
    
                f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]);
                if(canfold(i,j,k))
                {
    
    
                    int cnt=(len/(k-i+1));
                    f[i][j]=min(f[i][j],b[cnt]+f[i][k]+2);
                }
            }
        }
    }
    dfs(1,n);
}
signed main()
{
    
    
//    int t;
//    cin>>t;
//    while(t--)
        solve();
    return 0;
}


猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/109616420