Submission #986814

#	Time	Username	Problem	Language	Result	Execution time	Memory
986814		LOLOLO	Teams (CEOI11_tea)	C++17	100 / 100	849 ms	117948 KiB

tea

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
 
#define           f     first
#define           s     second
#define           pb    push_back
#define           ep    emplace
#define           eb    emplace_back
#define           lb    lower_bound
#define           ub    upper_bound
#define       all(x)    x.begin(), x.end()
#define      rall(x)    x.rbegin(), x.rend()
#define   uniquev(v)    sort(all(v)), (v).resize(unique(all(v)) - (v).begin())
#define     mem(f,x)    memset(f , x , sizeof(f))
#define        sz(x)    (int)(x).size()
#define  __lcm(a, b)    (1ll * ((a) / __gcd((a), (b))) * (b))
#define          mxx    *max_element
#define          mnn    *min_element
#define    cntbit(x)    __builtin_popcountll(x)
#define       len(x)    (int)(x.length())
 
const int N = 1e6 + 10;
vector <int> pos[N];
int a[N], dp[N], pr[N], n;
pair <int, int> b[N];
vector <int> val[N];
int trace[N];

int bin(int lim) {
    val[0].pb(0);
    for (int i = 1; i <= n; i++) {
        int t = dp[i];
        if (t == -1)
            continue;

        int p = lower_bound(all(val[t - 1]), i - lim) - val[t - 1].begin();
        if (p < sz(val[t - 1]) && b[i].f <= i - val[t - 1][p]) {
            trace[i] = val[t - 1][p];
            val[t].pb(i);
        }
    } 

    for (int i = 0; i <= n; i++)
        val[i].clear();

    return trace[n];
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin >> n;

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

    int sz = 0;
    for (int i = 1; i <= n; i++) {
        for (auto x : pos[i]) {
            sz++;
            b[sz] = {i, x};
        } 
    }

    dp[0] = pr[0] = 0;
    val[0].pb(0);

    for (int i = 1; i <= n; i++) {
        if (i < b[i].f) {
            dp[i] = -1;
        } else {
            dp[i] = dp[pr[i - b[i].f]] + 1;
        }

        pr[i] = dp[i] > dp[pr[i - 1]] ? i : pr[i - 1];
    }

    int l = 1, r = n, m, ans = n;
    while (l <= r) {
        mem(trace, -1);
        trace[n] = -1;
        m = (l + r) / 2;
        if (bin(m) >= 0) {
            ans = m;
            r = m - 1;
        } else {
            l = m + 1;
        }
    }

    cout << dp[n] << '\n';

    mem(trace, -1);
    bin(ans);

    vector < vector <int>> save;

    for (int i = n; i >= 1; i--) {
        assert(trace[i] != -1);
        cout << i - trace[i] << " ";
        for (int j = i; j > trace[i]; j--) {
            cout << b[j].s << " ";
        }

        i = trace[i] + 1;
        cout << '\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