Submission #1382348


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 pref[N], dp[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 >> pref[i];
        pref[i] += pref[i - 1];
    }
    for (int i = 1; i <= n; i++) {
        if (i < k) {
            continue;
        }
        dp[i] = dp[i - 1];
        ll val = pref[i] - pref[i - k];
        for (int j = i - k + 1; j <= i; j++) {
            if (i < n) val = max(val, pref[j] - pref[i - k]);
            if (i > k) val = max(val, pref[i] - pref[j - 1]);
        }
        dp[i] = max(dp[i], dp[i - k] + val);
    }
    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 1164 Byte
Status WA
Exec Time 2103 ms
Memory 1792 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 400
Status
AC × 4
AC × 14
WA × 2
TLE × 10
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 AC 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 1792 KB
1_04.txt AC 44 ms 1792 KB
1_05.txt AC 48 ms 1792 KB
1_06.txt AC 42 ms 1792 KB
1_07.txt AC 42 ms 1792 KB
1_08.txt AC 43 ms 1024 KB
1_09.txt AC 42 ms 1024 KB
1_10.txt WA 103 ms 1024 KB
1_11.txt TLE 2103 ms 1152 KB
1_12.txt TLE 2103 ms 1152 KB
1_13.txt TLE 2103 ms 1152 KB
1_14.txt WA 638 ms 1024 KB
1_15.txt TLE 2103 ms 1152 KB
1_16.txt TLE 2103 ms 1280 KB
1_17.txt TLE 2103 ms 1280 KB
1_18.txt TLE 2103 ms 1152 KB
1_19.txt TLE 2103 ms 1152 KB
1_20.txt TLE 2103 ms 1152 KB
1_21.txt TLE 2103 ms 1152 KB