oj.uz

Submission #674643

# Submission time Handle Problem Language Result Execution time Memory
674643 2022-12-25T14:46:17 Z tamthegod Teams (CEOI11_tea) C++14
50 / 100
 28 ms 536 KB 
// Make the best become better
// No room for laziness
#include<bits/stdc++.h>

#define int long long
#define pb push_back
#define fi first
#define se second
using namespace std;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int maxN = 5000 + 5;
const int mod = 1e9 + 7;
const ll oo = 1e18;
int n, a[maxN], id[maxN];
int f[maxN], trace[maxN];
int pos[maxN];
int dp[maxN];
void ReadInput()
{
    cin >> n;
    for(int i=1; i<=n; i++)
        cin >> a[i];
}
bool chk(int val)
{
    memset(dp, -3, sizeof dp);
    dp[0] = 0;
    for(int i=1; i<=n; i++)
    {
        int t = id[i];
        for(int j=i-a[t]+1; j>=max(1ll, i-val+1); j--)
        {
            if(dp[i] < dp[j - 1] + 1)
            {
                dp[i] = dp[j - 1] + 1;
            }
        }
    }
    return dp[n] == f[n];
}
void Print(int val)
{
    memset(f, -3, sizeof f);
    f[0] = 0;
    for(int i=1; i<=n; i++)
    {
        int t = id[i];
        for(int j=i-a[t]+1; j>=max(1ll, i-val+1); j--)
        {
            if(f[i] < f[j - 1] + 1)
            {
                f[i] = f[j - 1] + 1;
                trace[i] = j;
            }
        }
    }
    cout << f[n] << '\n';
    int i = n;
    while(i)
    {
        cout << i - trace[i] + 1 << " ";
        for(int j=i; j>=trace[i]; j--)
            cout << id[j] << " ";
        i = trace[i] - 1;
        cout << '\n';
    }
}
void Solve()
{
    iota(id + 1, id + n + 1, 1);
    sort(id + 1, id + n + 1, [](int i, int j)
    {
        return a[i] < a[j];
    });
    memset(f, -3, sizeof f);
    f[0] = 0;
    for(int i=1; i<=n; i++)
    {
        int t = id[i];
        for(int j=i-a[t]+1; j>=1; j--)
        {
            if(f[i] < f[j - 1] + 1)
            {
                f[i] = f[j - 1] + 1;
                trace[i] = j;
            }
        }
    }
    int low = a[id[n]], high = n, mid;
   // cout << chk(6);return;
    while(low <= high)
    {
        mid = (low + high) / 2;
        if(chk(mid)) high = mid - 1;
        else low = mid + 1;
    }
    //cout << low;return;
    Print(low);
}
int32_t main()
{
   // freopen("x.inp", "r", stdin);
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    ReadInput();
    Solve();
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 1 ms 332 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 1 ms 328 KB Output is correct
3 Correct 1 ms 340 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 23 ms 516 KB Output is correct
2 Correct 26 ms 536 KB Output is correct
3 Correct 28 ms 468 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 27 ms 468 KB Output is correct
2 Correct 22 ms 468 KB Output is correct

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 468 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #7
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 468 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #8
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 468 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 468 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #10
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 1 ms 468 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -