Submission #1269834

#	Time	Username	Problem	Language	Result	Execution time	Memory
1269834		yanb	Homecoming (BOI18_homecoming)	C++20	0 / 100	1094 ms	47428 KiB

homecoming

#include "homecoming.h"
#include <bits/stdc++.h>

using namespace std;

#define int long long

int solves(int n, int k, signed *A, signed *B, int shift) {
    vector<int> a(2 * n), b(2 * n);
    for (int i = 0; i < n; i++) {
        a[i] = a[i + n] = A[(i + shift) % n];
        b[i] = b[i + n] = B[(i + shift) % n];
    }

    vector<int> pa(2 * n + 1), pb(2 * n + 1);
    for (int i = 0; i < 2 * n; i++) {
        pa[i + 1] = pa[i] + a[i];
        pb[i + 1] = pb[i] + b[i];
    }

    vector<int> dp(n + 1, -1e17), dpm(2 * n + 1, -1e17);
    dp[0] = -pb[k - 1];
    dpm[k] = dp[0];
    int ans = dp[0];
    for (int i = 0; i < n - k; i++) {
        dp[i + 1] = max({dp[i] + a[i] - b[i + k - 1], dpm[i] + a[i] - pb[i + k] + pb[i]});
        dpm[i + k + 1] = max(dpm[i + k], dp[i + 1]);
        ans = max(ans, dp[i + 1]);
    }

    /*for (int x : b) cout << x << " ";
    cout << "\n";
    for (int x : dp) cout << x << " ";
    cout << "\n\n";*/

    return ans;
}

int solve(signed n, signed k, signed *A, signed *B) {
    int ans = accumulate(A, A + n, 0ll) - accumulate(B, B + n, 0ll);
    for (int i = 0; i < n; i++) {
        ans = max(ans, solves(n, k, A, B, i));
    }
    return ans;
}

• Subtask 1: 0 / 13
• Subtask 2: 0 / 18
• Subtask 3: 0 / 31
• Subtask 4: 0 / 38