Submission #554671

#	Time	Username	Problem	Language	Result	Execution time	Memory
554671		kwongweng	Bali Sculptures (APIO15_sculpture)	C++17	100 / 100	257 ms	16032 KiB

sculpture

/*
Solution for APIO 2015 - Bali Sculpture
Tags : 
*/

#include <bits/stdc++.h>
using namespace std;
 
#pragma GCC target ("avx2")
#pragma GCC optimization ("Ofast")
#pragma GCC optimization ("unroll-loops")
 
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef long double ld;
typedef pair<ll, ll> pll;
#define FOR(i, a, b) for(int i = a; i < b; i++)
#define ROF(i, a, b) for(int i = a; i >= b; i--)
#define ms memset
#define pb push_back
#define fi first
#define se second
 
ll MOD = 1000000007;
 
ll power(ll base, ll n){
	if (n == 0) return 1;
	if (n == 1) return base;
	ll halfn = power(base, n/2);
	if (n % 2 == 0) return (halfn * halfn) % MOD;
	return (((halfn * halfn) % MOD) * base) % MOD;
}
 
ll inverse(ll n){
	return power(n, MOD-2);
}
 
ll add(ll a, ll b){
	return (a+b) % MOD;
}
 
ll mul(ll a, ll b){
	a %= MOD;
	return (a*b) % MOD;
}
 
ll gcd(ll a, ll b){
    if (a == 1) return 1;
    if (a == 0) return b;
    return gcd(b%a, a);
}
 
void solve(){
    int n, a, b; cin >> n >> a >> b;
    vector<ll> y(n+1); FOR(i,1,n+1) cin >> y[i];
    vector<ll> p(n+1); FOR(i,1,n+1) p[i] = p[i-1] + y[i];
    int max_bit = 0;
    vector<ll> p2(51);
    p2[0]=1;
    FOR(i,1,51){
        p2[i] = p2[i-1] + p2[i-1];
    }
    while (p2[max_bit] <= p[n]){
        max_bit++;
    }
    max_bit--;
    if (a==1){
        int mat[n+1][n+2];
        ms(mat, 0, sizeof(mat));
        FOR(i,1,n+1){
            FOR(j,i+1,n+2){
                mat[i][j]=1;
            }
        }
        ll cur = 0;
        ROF(bit,max_bit,0){
            cur += p2[bit];
            FOR(i,1,n+1){
                FOR(j,i+1,n+2){
                    ll val = p[j-1]-p[i-1];
                    if (val&cur){
                        mat[i][j]=0;
                    }
                }
            }
            vi dp(n+2, MOD);
            dp[1]=0;
            FOR(i,1,n+1){
                FOR(j,i+1,n+2){
                    if (mat[i][j] == 1 && dp[i] != MOD){
                        dp[j]=min(dp[j],dp[i]+1);
                    }
                }
            }
            if (dp[n+1] > b){
                cur -= p2[bit];
            }
            FOR(i,1,n+1){
                FOR(j,i+1,n+2){
                    mat[i][j]=1;
                }
            }
        }
        cout << p2[max_bit+1] - 1 - cur;
        return;
    }
    int mat[n+1][n+2];
    FOR(i,1,n+1){
        FOR(j,i+1,n+2){
            mat[i][j] = 1;
        }
    }
    ll cur = 0;
    ROF(bit,max_bit,0){
        cur += p2[bit];
        FOR(i,1,n+1){
            FOR(j,i+1,n+2){
                ll val = p[j-1]-p[i-1];
                if (val&cur){
                    mat[i][j]=0;
                }
            }
        }
        int dist[n+2][n+1];
        ms(dist,-1,sizeof(dist));
        dist[1][0]=1;
        FOR(i,1,n+1){
            FOR(j,i+1,n+2){
                if (mat[i][j] == 1){
                    FOR(k,0,n){
                        if (dist[i][k] != -1){
                            if (dist[j][k+1] == -1){
                                dist[j][k+1] = 1;
                            }
                        }
                    }
                }
            }
        }
        bool sol = false;
        FOR(i,a,b+1){
            if (dist[n+1][i] == 1){
                sol = true; break;
            }
        }
        if (!sol){
            cur -= p2[bit];
        }
        FOR(i,1,n+1){
            FOR(j,i+1,n+2){
                mat[i][j] = 1;
            }
        }
    }
    cout << p2[max_bit+1] - cur - 1;
}
 
int main() {
    ios::sync_with_stdio(false);
    int TC = 1;
    //cin >> TC;
    FOR(i, 1, TC+1){
        //cout << "Case #" << i << ": ";
        solve();
    }
}

Compilation message (stderr)

sculpture.cpp:10: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
   10 | #pragma GCC optimization ("Ofast")
      | 
sculpture.cpp:11: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
   11 | #pragma GCC optimization ("unroll-loops")
      |

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