Submission #1073911

#	Time	Username	Problem	Language	Result	Execution time	Memory
1073911		ngrace	Bali Sculptures (APIO15_sculpture)	C++14	100 / 100	95 ms	1236 KiB

sculpture

#include <bits/stdc++.h>
using namespace std;
#define v vector
#define ll long long
#define ii pair<int,int>
#define fi first
#define se second
#define on(i,mask) ((bool)((1<<(i)) & (mask)))

int main(){
    int N,A,B;
    cin>>N>>A>>B;
    v<ll> Y(N+1,0);
    v<ll> psum(N+1, 0);
    ll maxY = 0;
    for(int i=1; i<=N; i++){
        cin>>Y[i];
        maxY = max(maxY, Y[i]);
        psum[i] = psum[i-1]+Y[i];
    }

    if(N<=20){//sub1
        ll ans = 1e18;
        for(int div = (1<<N); div<(1<<(N+1)); div++){
            int g = 1;
            int l=0;
            ll res = 0;
            for(int i=1; i<=N; i++){
                if(i!=1 && on(i-2,div)!=on(i-1,div)){
                    g++;
                    res = res | (psum[i-1] - psum[l]);
                    l=i-1;
                }
            }
            res = res | (psum[N]-psum[l]);

            if(A<=g && g<=B) ans = min(ans, res);
        }
        cout<<ans<<endl;
    }
    else if(N<=50 && B<=20 && maxY<=10){//sub2
        v<v<v<bool>>> dp(B+1, v<v<bool>>(N+1, v<bool>(maxY*N+1, false)));
        dp[0][0][0] = true;
        for(int b=0; b<B; b++){
            for(int n=0; n<=N; n++){
                for(int o=0;o<=maxY*N;o++){
                    if(dp[b][n][o]){//can cover first n in b groups with or sum o
                        for(int i=n+1; i<=N; i++){
                            dp[b+1][i][o | (psum[i]-psum[n])] = true;
                        }
                    }
                }
            }
        }

        int ans = maxY*N;
        for(int b=A; b<=B; b++){
            for(int o=0; o<=ans; o++){
                if(dp[b][N][o]){
                    ans=o;
                    break;
                }
            }
        }
        cout<<ans<<endl;
    }
    else if(N<=100 && A==1 && maxY<=20){//sub3
        v<v<int>> dp(N+1, v<int>(maxY*N+1, 1e9));
        dp[0][0] = 0;
        for(int n=0; n<=N; n++){
            for(int o=0;o<=maxY*N;o++){
                if(dp[n][o] !=1e9){//min groups to cover first n with or sum o
                    for(int i=n+1; i<=N; i++){
                        dp[i][o | (psum[i]-psum[n])] = min(dp[i][o | (psum[i]-psum[n])], dp[n][o]+1);
                    }
                }
            }
        }

        int ans = maxY*N;
        for(int o=0; o<=ans; o++){
            if(dp[N][o]<=B){
                ans=o;
                break;
            }
        }
        cout<<ans<<endl;
    }
    else if(N<=100){//sub4
        ll is0 = 0;
        for(int bit=40; bit>=0; bit--){
            v<v<bool>> dp(B+1, v<bool>(N+1, false));
            ll test = is0 | (1ll<<bit);
            dp[0][0] = true;
            for(int b=0; b<B; b++){
                for(int n=0; n<=N; n++){
                    if(dp[b][n]){//can cover first n in b groups while keeping all on bits in is0 zero.
                        for(int i=n+1; i<=N; i++){
                            ll s = psum[i] - psum[n];
                            if((s & test) == 0) dp[b+1][i] = true;
                        }
                    }
                }
            }

            for(int b=A; b<=B; b++){
                if(dp[b][N]){
                    is0 |= (1ll<<bit);
                    break;
                }
            }
        }
        cout<<((1ll<<41) - is0 - 1)<<endl;
    }
    else if(N<=2000 && A==1){//sub 5
        ll is0 = 0;
        for(int bit=60; bit>=0; bit--){
            v<int> dp(N+1, 1e9);
            ll test = is0 | (1ll<<bit);
            dp[0] = 0;
            for(int n=0; n<=N; n++){
                if(dp[n] != 1e9){//min groups to cover first n while keeping all on bits in is0 zero.
                    for(int i=n+1; i<=N; i++){
                        ll s = psum[i] - psum[n];
                        if((s & test) == 0) dp[i] = min(dp[i], dp[n]+1);
                    }
                }
            }

            if(dp[N]<=B){
                is0 |= (1ll<<bit);
            }
        }
        cout<<((1ll<<61) - is0 - 1)<<endl;
    }
}

• Subtask 1: 9 / 9
• Subtask 2: 16 / 16
• Subtask 3: 21 / 21
• Subtask 4: 25 / 25
• Subtask 5: 29 / 29