Submission #1382479


Source Code Expand

#include <bits/stdc++.h>

#define pii pair <int, int>

#define mp make_pair

#define f first
#define s second

#define pb push_back

using namespace std;

typedef unsigned long long ull;
typedef long long ll;
typedef long double ld;

const int N = 1e5 + 10;
const int NN = 20;
const int INF = 1e9 + 7;

int n, k;
ll a[N], dp[N], pref[N], pos[N];

int main() {
    #define fn "balls"
    #ifdef witch
        freopen("input.txt", "r", stdin);
        freopen("output.txt", "w", stdout);
    #else
//        freopen(fn".in", "r", stdin);
//        freopen(fn".out", "w", stdout);
    #endif
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        pref[i] = pref[i - 1] + a[i];
    }
    for (int i = 1; i <= n; i++) {
        if (i < k) {
            dp[i] = dp[i - 1];
            if (n - i >= k)
                dp[i] = max(dp[i], pref[i]);
            continue;
        }
        if (i > n - k && i - 1 >= k) {
            dp[i] = max(dp[i], dp[i - 1] + pref[n] - pref[i - 1]);
        }
        if (i > k && i < n - k + 1) {
            dp[i] = max(dp[i], dp[i - 1] + a[i]);
        }
        dp[i] = max(dp[i], dp[i - 1]);
        dp[i] = max(dp[i], dp[i - k] + pref[i] - pref[i - k]);
    }
    cout << dp[n];
    return 0;
}

Submission Info

Submission Time
Task B - Contiguous Repainting
User Nurlykhan
Language C++14 (GCC 5.4.1)
Score 0
Code Size 1312 Byte
Status WA
Exec Time 48 ms
Memory 2560 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 400
Status
AC × 3
WA × 1
AC × 13
WA × 13
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 0_03.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt
Case Name Status Exec Time Memory
0_00.txt AC 1 ms 256 KB
0_01.txt AC 1 ms 256 KB
0_02.txt AC 1 ms 256 KB
0_03.txt WA 1 ms 256 KB
1_00.txt AC 1 ms 256 KB
1_01.txt AC 1 ms 256 KB
1_02.txt AC 1 ms 256 KB
1_03.txt AC 18 ms 2560 KB
1_04.txt AC 45 ms 2560 KB
1_05.txt AC 48 ms 2560 KB
1_06.txt AC 42 ms 2560 KB
1_07.txt AC 42 ms 2560 KB
1_08.txt AC 43 ms 2560 KB
1_09.txt AC 42 ms 2560 KB
1_10.txt WA 42 ms 2560 KB
1_11.txt WA 46 ms 2560 KB
1_12.txt WA 43 ms 2560 KB
1_13.txt WA 42 ms 2560 KB
1_14.txt WA 41 ms 2560 KB
1_15.txt WA 44 ms 2560 KB
1_16.txt WA 43 ms 2560 KB
1_17.txt WA 41 ms 2560 KB
1_18.txt WA 43 ms 2560 KB
1_19.txt WA 41 ms 2560 KB
1_20.txt WA 43 ms 2560 KB
1_21.txt WA 42 ms 2560 KB