Submission #78029

#	Time	Username	Problem	Language	Result	Execution time	Memory
78029		xiaowuc1	Bali Sculptures (APIO15_sculpture)	C++14	100 / 100	184 ms	1020 KiB

sculpture

#include <bits/stdc++.h>

/*
unsigned seed1 = std::chrono::system_clock::now().time_since_epoch().count();
mt19937 g1.seed(seed1);

ios_base::sync_with_stdio(false);
cin.tie(NULL);
*/
using namespace std;

const double PI = 2 * acos(0);

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<int, ll> pill;
typedef pair<ll, ll> pll;
typedef long double ld;
typedef vector<vector<ll>> matrix;

ll pref[2001];

int n, lhs, rhs;
bool valid[2001];
bool nextValid[2001];

int dp[2001];

bool fastcan(ll mask) {
  for(int i = 1; i <= n; i++) {
    dp[i] = 1 << 25;
  }
  for(int i = 0; i < n; i++) {
    for(int j = i+1; j <= n; j++) {
      ll vv = (pref[j] - pref[i]) & mask;
      if(vv == pref[j] - pref[i]) {
        dp[j] = min(dp[j], dp[i]+1);
      }
    }
  }
  return dp[n] <= rhs;
}

bool can(ll mask) {
  if(lhs == 1) return fastcan(mask);
  memset(valid, 0, sizeof(valid));
  valid[0] = true;
  for(int x = 1; x <= rhs; x++) {
    memset(nextValid, 0, sizeof(nextValid));
    for(int a = 0; a <= n; a++) {
      if(!valid[a]) continue;
      for(int b = a+1; b <= n; b++) {
        ll vv = pref[b] - pref[a];
        nextValid[b] |= (vv&mask)==vv;
      }
    }
    if(x >= lhs && nextValid[n]) return true;
    memcpy(valid, nextValid, sizeof(valid));
  }
  return false;
}

int main() {
  scanf("%d%d%d", &n, &lhs, &rhs);
  for(int i = 0; i < n; i++) {
    scanf("%lld", &pref[i+1]);
    pref[i+1] += pref[i];
  }
  ll ret = 1;
  while(ret < pref[n]) {
    ret |= ret << 1;
  }
  for(int i = 45; i >= 0; i--) {
    ll ans = ret & ~(1LL << (i));
    if(ans == ret) continue;
    if(can(ans)) {
      ret = ans;
    }
  }
  printf("%lld\n", ret);
}

Compilation message (stderr)

sculpture.cpp: In function 'int main()':
sculpture.cpp:65:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d%d", &n, &lhs, &rhs);
   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~
sculpture.cpp:67:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld", &pref[i+1]);
     ~~~~~^~~~~~~~~~~~~~~~~~~~

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