Submission #545362

#	Time	Username	Problem	Language	Result	Execution time	Memory
545362		leaked	Copy and Paste 3 (JOI22_copypaste3)	C++14	57 / 100	889 ms	53592 KiB

copypaste3

#include <bits/stdc++.h>

#define f first
#define s second
#define m_p make_pair
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define vec vector
#define pb push_back
#define sz(x) (int)(x).size()
#define pw(x) (1LL<<(x))
#define fast_resp ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef long double ld;
typedef pair<int,ll> pil;
template<class T> bool umin(T &a,const T &b){return (a>b?a=b,1:0);}
template<class T> bool umax(T &a,const T &b){return (a<b?a=b,1:0);}
const ll inf=1e18;
const int N=1e3+3;
ll dp[2][N][N];
///l,r,prefix
//int pref[N][N];
vec<int> zet[N];
vec<int> z_func(string s){
    int l=0,r=-1;
    vec<int> zt(sz(s),0);
    for(int i=1;i<sz(s);i++){
        zt[i]=(i>r?0:min(r-i+1,zt[i-l]));
        while(i+zt[i]<sz(s) && s[i+zt[i]]==s[zt[i]]) ++zt[i];
        if(i+zt[i]-1>r){
            r=i+zt[i]-1;
            l=i;
        }
    }
    return zt;
}
signed main(){
    fast_resp;
    int n;
    string s;
    cin>>n>>s;
    int a,b,c;
    cin>>a>>b>>c;
    for(int t=0;t<2;t++){
        for(int j=0;j<N;j++){
            for(int k=0;k<N;k++)
                dp[t][j][k]=inf;
        }
    }
    for(int i=0;i<n;i++){
        zet[i]=z_func(s.substr(i,n-i)+'#'+s.substr(i,n-i));
//        for(int j=i;j<n;j++){
//            cout<<s.substr(i,n-i)<<' '<<s.substr(j,n-j)<<' '<<zet[i][j-i+n-i+1]<<'\n';
//        }
//        for(int j=i+1;j<n;j++){
//            int jt=pref[i][j-1];
//            while(jt && s[j]!=s[jt+i]) jt=pref[i][jt-1];
//            pref[i][j]=jt+(s[jt+i]==s[j]);
//        }
    }
    for(int l=n-1;l>=0;l--){
        int nl=(l%2)^1;
        for(int r=max(l-1,0);r<n;r++){
            for(int k=0;k<=r-l+2;k++)
                dp[nl][r][k]=inf;
        }
        int lt=l%2;
        for(int r=l;r<n;r++){
            if(l==r) dp[lt][r][0]=a;
            for(int k=0;k<=r-l+1;k++){
//                cout<<"HEY "<<r-k+2+n-l<<' '<<sz(zet[l])<<endl;
                if(r-k+1>=l+k && zet[l][r-k+2+n-l-l]>=k){
//                    assert((r-k+2+n-l)<sz(zet[l]));

//                    if(zet[l][r-k+2+n-l-l]>=k){
//                    assert((r-k+2+n-l)<sz(zet[l]));
                    umin(dp[lt][r][k],dp[lt][r-k][k]+c);
//                    }
                }
                umin(dp[lt][r][r-l+1],dp[lt][r][k]+b+c);
                if(l)
                    umin(dp[nl][r][0],dp[lt][r][k]+a);
                if(r+1<n)
                    umin(dp[lt][r+1][k],dp[lt][r][k]+a);
                umin(dp[lt][r][0],dp[lt][r][k]);
            }
        }
    }
//    cout<<dp[2][4][3]<<endl;
    ll ans=inf;
    for(int i=0;i<=n;i++)
        umin(ans,dp[0][n-1][i]);
    cout<<ans;
    return 0;
}
/*

*/

• Subtask 1: 1 / 1
• Subtask 2: 0 / 5
• Subtask 3: 14 / 14
• Subtask 4: 10 / 10
• Subtask 5: 32 / 32
• Subtask 6: 0 / 38