Submission #861161

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
861161	2023-10-15T14:00:29 Z	sleepntsheep	Feast (NOI19_feast)	C++17	51 / 100	34 ms	4700 KB

feast

#include <cstdio>
#include <climits>
#include <functional>
#include <numeric>
#include <cstring>
#include <cassert>
#include <string>
#include <deque>
#include <vector>
#include <map>
#include <queue>
#include <algorithm>
#include <iostream>
#include <utility>
using namespace std;
using ll=long long;

#define N 300005
#define ALL(x) x.begin(), x.end()

int n, k, a[N];
long long p[N];

int main()
{
    cin.tie(0)->sync_with_stdio(0);
    cin >> n >> k;
    for (int i = 1; i <= n; ++i) cin >> a[i], p[i] = p[i-1] + a[i];

    if (count_if(a+1, a+1+n,[](int x){ return x<0; }) == 0) return cout << accumulate(a+1,a+1+n, 0ll), 0;

    if (count_if(a+1, a+1+n,[](int x){ return x<0; }) == 1)
    {
        int x=0;
        for (int i = 1; i <= n; ++i) if (a[i] < 0) x = i;
        if (k == 1) return cout << max({p[x-1], p[n] - p[x], p[n]}), 0;
        else return cout << p[n] - a[x], 0;
    }

    if (k == 1)
    {
        long long ans = 0, mn = 0;
        for (int i = 1; i <= n; ++i) ans = max(ans, p[i] - mn), mn = min(mn, p[i]);
        return cout << ans, 0;
    }

    if (n <= 300)
    {
        long long dp[305][305]{0};
        for (int i = 1; i <= n; ++i)
            for (int j = 1; j <= k; ++j)
            {
                dp[i][j] = max({dp[i-1][j], dp[i][j], dp[i][j-1]});
                for (int l = 0; l < i; ++l) dp[i][j] = max(dp[i][j], dp[l][j-1] + p[i] - p[l]);
            }
        return cout << dp[n][k], 0;
    }

    if (n <= 3000)
    {
        vector<vector<vector<long long>>> dp(2, vector<vector<long long>>(3005, vector<long long>(2)));

        long long ans = 0;
        for (int i = 1; i <= n; ++i)
        {
            for (int j = 1; j <= k; ++i)
            {
                dp[i&1][j][0] = max(dp[!(i&1)][j][0], dp[!(i&1)][j][1]);
                dp[i&1][j][1] = max({a[i] + dp[!(i&1)][j-1][0], a[i] + dp[!(i&1)][j][1]});
                ans = max({ans, dp[i&1][j][0], dp[i&1][j][1]});
            }
        }
        return cout << ans, 0;
    }

    //assert(0);
#if 0
    {
        function<void(int, int, int, int, int)> compute = [&](int i, int l, int r, int optl, int optr)
        {
            if (l > r) return;
            int m = (l+r)/2;
            pair<long long, int> best = {LLONG_MIN, -1};
            for (int k = optl; k <= min(m, optr); ++k)
                best = max(best, {(dp[!(i&1)][k]) + p[m] - p[k], k});
            dp[i&1][m] = best.first;
            compute(i, l, m-1, optl, best.second), compute(i, m+1, r, best.second, optr);
        };

        long long ans = 0;
        for (int i = 1; i <= k; ++i)
        {
            compute(i, 1, n, 0, n-1);
            for (int i = 1; i <= n; ++i) dp[i&1][i] = max(dp[i&1][i], dp[i&1][i-1]);
            ans = max(ans, *max_element(dp[i&1], dp[i&1]+n+1));
        }
        return cout << ans,0;
    }
#endif

    return 0;
}

Subtask #1
4.0 / 4.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	24 ms	4444 KB	Output is correct
2	Correct	23 ms	4444 KB	Output is correct
3	Correct	23 ms	4700 KB	Output is correct
4	Correct	34 ms	4496 KB	Output is correct
5	Correct	22 ms	4576 KB	Output is correct
6	Correct	24 ms	4444 KB	Output is correct
7	Correct	22 ms	4444 KB	Output is correct
8	Correct	23 ms	4444 KB	Output is correct
9	Correct	22 ms	4436 KB	Output is correct
10	Correct	24 ms	4444 KB	Output is correct

Subtask #2
8.0 / 8.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	15 ms	4444 KB	Output is correct
2	Correct	16 ms	4700 KB	Output is correct
3	Correct	15 ms	4440 KB	Output is correct
4	Correct	15 ms	4416 KB	Output is correct
5	Correct	22 ms	4408 KB	Output is correct
6	Correct	15 ms	4444 KB	Output is correct
7	Correct	15 ms	4700 KB	Output is correct
8	Correct	25 ms	4684 KB	Output is correct
9	Correct	22 ms	4444 KB	Output is correct
10	Correct	15 ms	4440 KB	Output is correct

Subtask #3
18.0 / 18.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	34 ms	4440 KB	Output is correct
2	Correct	25 ms	4444 KB	Output is correct
3	Correct	25 ms	4608 KB	Output is correct
4	Correct	25 ms	4444 KB	Output is correct
5	Correct	25 ms	4640 KB	Output is correct
6	Correct	25 ms	4432 KB	Output is correct
7	Correct	25 ms	4440 KB	Output is correct
8	Correct	25 ms	4444 KB	Output is correct
9	Correct	25 ms	4572 KB	Output is correct
10	Correct	26 ms	4436 KB	Output is correct

Subtask #4
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	1116 KB	Output is correct
2	Correct	1 ms	1116 KB	Output is correct
3	Correct	1 ms	1112 KB	Output is correct
4	Correct	1 ms	1112 KB	Output is correct
5	Correct	1 ms	1116 KB	Output is correct
6	Correct	1 ms	1116 KB	Output is correct
7	Correct	1 ms	1116 KB	Output is correct
8	Correct	1 ms	1112 KB	Output is correct
9	Correct	1 ms	1116 KB	Output is correct
10	Correct	1 ms	1116 KB	Output is correct

Subtask #5
11.0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	1116 KB	Output is correct
2	Correct	1 ms	1116 KB	Output is correct
3	Correct	1 ms	1112 KB	Output is correct
4	Correct	1 ms	1112 KB	Output is correct
5	Correct	1 ms	1116 KB	Output is correct
6	Correct	1 ms	1116 KB	Output is correct
7	Correct	1 ms	1116 KB	Output is correct
8	Correct	1 ms	1112 KB	Output is correct
9	Correct	1 ms	1116 KB	Output is correct
10	Correct	1 ms	1116 KB	Output is correct
11	Correct	3 ms	1116 KB	Output is correct
12	Correct	3 ms	1192 KB	Output is correct
13	Correct	2 ms	1128 KB	Output is correct
14	Correct	3 ms	1116 KB	Output is correct
15	Correct	2 ms	1116 KB	Output is correct
16	Correct	2 ms	1116 KB	Output is correct
17	Correct	3 ms	1116 KB	Output is correct
18	Correct	1 ms	1116 KB	Output is correct
19	Correct	1 ms	1116 KB	Output is correct
20	Correct	1 ms	1116 KB	Output is correct

Subtask #6
0 / 20.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	1116 KB	Output is correct
2	Correct	1 ms	1116 KB	Output is correct
3	Correct	1 ms	1112 KB	Output is correct
4	Correct	1 ms	1112 KB	Output is correct
5	Correct	1 ms	1116 KB	Output is correct
6	Correct	1 ms	1116 KB	Output is correct
7	Correct	1 ms	1116 KB	Output is correct
8	Correct	1 ms	1112 KB	Output is correct
9	Correct	1 ms	1116 KB	Output is correct
10	Correct	1 ms	1116 KB	Output is correct
11	Correct	3 ms	1116 KB	Output is correct
12	Correct	3 ms	1192 KB	Output is correct
13	Correct	2 ms	1128 KB	Output is correct
14	Correct	3 ms	1116 KB	Output is correct
15	Correct	2 ms	1116 KB	Output is correct
16	Correct	2 ms	1116 KB	Output is correct
17	Correct	3 ms	1116 KB	Output is correct
18	Correct	1 ms	1116 KB	Output is correct
19	Correct	1 ms	1116 KB	Output is correct
20	Correct	1 ms	1116 KB	Output is correct
21	Runtime error	4 ms	2908 KB	Execution killed with signal 11
22	Halted	0 ms	0 KB	-

Subtask #7
0 / 29.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	24 ms	4444 KB	Output is correct
2	Correct	23 ms	4444 KB	Output is correct
3	Correct	23 ms	4700 KB	Output is correct
4	Correct	34 ms	4496 KB	Output is correct
5	Correct	22 ms	4576 KB	Output is correct
6	Correct	24 ms	4444 KB	Output is correct
7	Correct	22 ms	4444 KB	Output is correct
8	Correct	23 ms	4444 KB	Output is correct
9	Correct	22 ms	4436 KB	Output is correct
10	Correct	24 ms	4444 KB	Output is correct
11	Correct	15 ms	4444 KB	Output is correct
12	Correct	16 ms	4700 KB	Output is correct
13	Correct	15 ms	4440 KB	Output is correct
14	Correct	15 ms	4416 KB	Output is correct
15	Correct	22 ms	4408 KB	Output is correct
16	Correct	15 ms	4444 KB	Output is correct
17	Correct	15 ms	4700 KB	Output is correct
18	Correct	25 ms	4684 KB	Output is correct
19	Correct	22 ms	4444 KB	Output is correct
20	Correct	15 ms	4440 KB	Output is correct
21	Correct	34 ms	4440 KB	Output is correct
22	Correct	25 ms	4444 KB	Output is correct
23	Correct	25 ms	4608 KB	Output is correct
24	Correct	25 ms	4444 KB	Output is correct
25	Correct	25 ms	4640 KB	Output is correct
26	Correct	25 ms	4432 KB	Output is correct
27	Correct	25 ms	4440 KB	Output is correct
28	Correct	25 ms	4444 KB	Output is correct
29	Correct	25 ms	4572 KB	Output is correct
30	Correct	26 ms	4436 KB	Output is correct
31	Correct	1 ms	1116 KB	Output is correct
32	Correct	1 ms	1116 KB	Output is correct
33	Correct	1 ms	1112 KB	Output is correct
34	Correct	1 ms	1112 KB	Output is correct
35	Correct	1 ms	1116 KB	Output is correct
36	Correct	1 ms	1116 KB	Output is correct
37	Correct	1 ms	1116 KB	Output is correct
38	Correct	1 ms	1112 KB	Output is correct
39	Correct	1 ms	1116 KB	Output is correct
40	Correct	1 ms	1116 KB	Output is correct
41	Correct	3 ms	1116 KB	Output is correct
42	Correct	3 ms	1192 KB	Output is correct
43	Correct	2 ms	1128 KB	Output is correct
44	Correct	3 ms	1116 KB	Output is correct
45	Correct	2 ms	1116 KB	Output is correct
46	Correct	2 ms	1116 KB	Output is correct
47	Correct	3 ms	1116 KB	Output is correct
48	Correct	1 ms	1116 KB	Output is correct
49	Correct	1 ms	1116 KB	Output is correct
50	Correct	1 ms	1116 KB	Output is correct
51	Runtime error	4 ms	2908 KB	Execution killed with signal 11
52	Halted	0 ms	0 KB	-

Submission #861161

Compilation message

Subtask #1 4.0 / 4.0

Subtask #2 8.0 / 8.0

Subtask #3 18.0 / 18.0

Subtask #4 10.0 / 10.0

Subtask #5 11.0 / 11.0

Subtask #6 0 / 20.0

Subtask #7 0 / 29.0

Subtask #1
4.0 / 4.0

Subtask #2
8.0 / 8.0

Subtask #3
18.0 / 18.0

Subtask #4
10.0 / 10.0

Subtask #5
11.0 / 11.0

Subtask #6
0 / 20.0

Subtask #7
0 / 29.0