Submission #65415

#	Time	Username	Problem	Language	Result	Execution time	Memory
65415		Googal	Teams (CEOI11_tea)	C++14	100 / 100	637 ms	90800 KiB

tea

#include <algorithm>
#include <iostream>
#include <memory.h>
#include <bitset>

using namespace std;

const int NMAX = 1e6 + 1e2;
const int INF = 1e9;

int n;
int dp[NMAX];
int pr[NMAX], opt[NMAX], par[NMAX];

pair < int, int > a[NMAX];
bitset < NMAX > ok;

bool check(int x) {
  ok = 0;
  ok[0] = 1;

  for(int i = 1; i <= n; i++) {
    if(i - a[i].first < 0) {
      dp[i] = -INF;
    } else {
      dp[i] = pr[i - a[i].first] + 1;

      if(opt[i - a[i].first] >= i - x) {
        ok[i] = 1;
        par[i] = opt[i - a[i].first];
      }
    }

    pr[i] = pr[i - 1];
    opt[i] = opt[i - 1];

    if(ok[i] == 1 && pr[i] <= dp[i]) {
      opt[i] = i;
      pr[i] = dp[i];
    }
  }

  return (bool)ok[n];
}

void printsol(int x) {
  if(x == 0)
    return;

  int p1 = par[x];
  printsol(p1);

  cout << x - p1 << ' ';

  for(int i = p1 + 1; i <= x; i++)
    cout << a[i].second << ' ';
  cout << '\n';
}

void bin_src(int &le, int &ri) {
  while(le != ri) {
    int mid = (le + ri) / 2;
    if(check(mid) == false)
      le = mid + 1;
    else
      ri = mid;
  }
}

int main()
{
  ios::sync_with_stdio(false);

  cin >> n;
  for(int i = 1; i <= n; i++) {
    cin >> a[i].first;
    a[i].second = i;
  }

  sort(a + 1, a + n + 1);

  int le = 1;
  int ri = n;

  bin_src(le, ri);

  check(le);
  cout << dp[n] << '\n';

  printsol(n);
  return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10