Submission #851038

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
851038	2023-09-18T09:56:32 Z	Nhoksocqt1	Calvinball championship (CEOI15_teams)	C++17	0 / 100	393 ms	900 KB

teams

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

#define inf 0x3f3f3f3f
#define sz(x) int((x).size())
#define fi first
#define se second
typedef long long ll;
typedef pair<int, int> ii;

template<class X, class Y>
	inline bool maximize(X &x, const Y &y) {return (x < y ? x = y, 1 : 0);}
template<class X, class Y>
	inline bool minimize(X &x, const Y &y) {return (x > y ? x = y, 1 : 0);}

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int Random(int l, int r) {
    return uniform_int_distribution<int>(l, r)(rng);
}

const int MAXN = 10004;
const int MOD = 1000007;

ll dp[2][MAXN][2], dp2[2][MAXN];
int a[MAXN], pref[MAXN], nArr;

ll powermod(ll a, int exponent) {
    ll res(1);
    while(exponent > 0) {
        if(exponent & 1)
            res = res * a % MOD;

        a = a * a % MOD;
        exponent >>= 1;
    }

    return res;
}

void process() {
    cin >> nArr;
    for (int i = 1; i <= nArr; ++i) {
        cin >> a[i];
        pref[i] = max(pref[i - 1], a[i]);
    }

    dp[1][1][0] = dp2[1][1] = 1;
    for (int i = 2; i <= nArr; ++i) {
        int cur(i & 1), pre(!cur);
        for (int j = 1; j <= pref[i - 1]; ++j) {
            for (int smaller = 0; smaller < 2; ++smaller) {
                if(!dp[pre][j][smaller])
                    continue;

                if(!smaller) {
                    dp[cur][j][1] = (dp[cur][j][1] + dp[pre][j][smaller] * min(j, a[i] - 1)) % MOD;
                    dp[cur][j][0] = (dp[cur][j][0] + dp[pre][j][smaller] * (j >= a[i])) % MOD;
                } else {
                    dp[cur][j][smaller] = (dp[cur][j][smaller] + dp[pre][j][smaller] * j) % MOD;
                }

                if((smaller || j + 1 <= a[i]) && (dp[cur][j + 1][smaller | (j + 1 < a[i])] += dp[pre][j][smaller]) >= MOD)
                    dp[cur][j + 1][smaller | (j + 1 < a[i])] -= MOD;

                dp[pre][j][smaller] = 0;
            }
        }

        for (int j = 1; j <= min(i - 1, pref[nArr] - 1); ++j) {
            if(!dp2[pre][j])
                continue;

            dp2[cur][j] = (dp2[cur][j] + dp2[pre][j] * j) % MOD;
            if((dp2[cur][j + 1] += dp2[pre][j]) >= MOD)
                dp2[cur][j + 1] -= MOD;

            dp2[pre][j] = 0;
        }
    }


    ll res = dp2[nArr & 1][pref[nArr] - 1] + dp[nArr & 1][pref[nArr]][0] + dp[nArr & 1][pref[nArr]][1];
    cout << res % MOD << '\n';
}

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

    process();
    return 0;
}

Subtask #1
0 / 10.0

#	Verdict	Memory	Grader output
1	Correct	348 KB	Output is correct
2	Correct	348 KB	Output is correct
3	Correct	600 KB	Output is correct
4	Correct	348 KB	Output is correct
5	Incorrect	348 KB	Output isn't correct
6	Halted	0 KB	-

Subtask #2
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	348 KB	Output is correct
2	Incorrect	0 ms	344 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #3
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1 ms	344 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #4
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1 ms	348 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #5
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	0 ms	348 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #6
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1 ms	348 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #7
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	1 ms	468 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #8
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	393 ms	900 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #9
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	4 ms	348 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #10
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	9 ms	348 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Submission #851038

Compilation message