Submission #497398

#	Time	Username	Problem	Language	Result	Execution time	Memory
497398		SirCovidThe19th	Teams (CEOI11_tea)	C++17	100 / 100	383 ms	31668 KiB

tea

#include <bits/stdc++.h>
using namespace std; 

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0); 
    int n; cin >> n; pair<int, int> A[n + 1]; 
    for (int i = 1; i <= n; i++) cin >> A[i].first, A[i].second = i;
    sort(A + 1, A + n + 1);

    auto maxTeams = [&](int mxsz, bool get){
        int dp[n + 1], sz[n + 1], lft[n + 1]; 
        memset(dp, -0x3f, sizeof(dp)); dp[0] = lft[0] = 0;

        for (int i = 1; i <= n; i++){
            int l = max(i - mxsz, 0), r = i - A[i].first;
            if (r >= 0 and l <= lft[r]){
                dp[i] = dp[lft[r]] + 1; sz[i] = i - lft[r];
                if (i != n and dp[lft[i - 1]] > dp[i]) dp[i] = -1e9; //not optimal to transition from i
            }
            lft[i] = (dp[i] >= 0) ? i : lft[i - 1];
        }
        if (get){
            cout<<dp[n]<<"\n";
            for (int pos = n; pos; pos -= sz[pos]){
                cout<<sz[pos]<<" ";
                for (int i = pos; i > pos - sz[pos]; i--) cout<<A[i].second<<" ";
                cout<<"\n";
            }
        }
        return dp[n];
    };    
    int L = 1, H = n, mx = maxTeams(1e9, 0);
    while (L < H){
        int M = (L + H) / 2;
        (maxTeams(M, 0) == mx) ? H = M : L = M + 1;
    }
    maxTeams(H, 1);
}

• 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