제출 #1343564

#제출 시각아이디문제언어결과실행 시간메모리
1343564biankHomecoming (BOI18_homecoming)C++20
31 / 100
651 ms266596 KiB
#include <bits/stdc++.h>
#include "homecoming.h"
 
using namespace std;
 
#define forsn(i, s, n) for (int i = int(s); i < int(n); i++)
#define forn(i, n) forsn(i, 0, n)
#define dforsn(i, s, n) for (int i = int(n) - 1; i >= int(s); i--)
#define dforn(i, n) dforsn(i, 0, n)
 
using vi = vector<int>;
using ii = pair<int, int>;
using vii = vector<ii>;
using ll = long long;
using ld = long double;
using vll = vector<ll>;
using vb = vector<bool>;
using pll = pair<ll, ll>;

#define sz(x) int(x.size())
#define all(x) begin(x), end(x)

#define pb push_back
#define eb emplace_back

#define fst first
#define snd second

const ll INF = 1e18;

using mat = array<array<ll, 2>, 2>;

mat mul(const mat &a, const mat &b) {
    mat c = {-INF, -INF, -INF, -INF};
    forn(i, 2) forn(k, 2) forn(j, 2) {
        c[i][j] = max(c[i][j], a[i][k] + b[k][j]);
    }
    return c;
}

ll solve(int N, int K, int *A, int *B) {
    ll ret = 0;
    forn(i, N) ret += A[i] - B[i];
    int s = 2 * N;
    vector<mat> st(2 * s, {0, -INF, -INF, 0});
    ll window = 0;
    forn(i, K) window += B[i];
    forn(i, 2 * N) {
        st[i + s] = {0, -window + A[i % N], 0, -B[(i + K - 1) % N] + A[i % N]};
        window -= B[i % N];
        window += B[(i + K) % N];
    }
    dforsn(i, 1, s) st[i] = mul(st[2 * i], st[2 * i + 1]);
    forn(start, N) {
        mat leftRet = {0, -INF, -INF, 0};
        mat rightRet = {0, -INF, -INF, 0};
        for (int l = start + s, r = start + N + s; l < r; l /= 2, r /= 2) {
            if (l & 1) leftRet = mul(leftRet, st[l++]);
            if (r & 1) rightRet = mul(st[--r], rightRet);
        }
        mat curr = mul(leftRet, rightRet);
        ret = max({ret, curr[0][0], curr[0][1]});
    }
    st.clear();
    return ret;
}

/*int main() {
    ios::sync_with_stdio(0);         
    cin.tie(0); cout.tie(0);
    
    int a[] = {40, 80, 100};
    int b[] = {140, 0, 20};
    cout << solve(3, 2, a, b) << "\n";
    
    return 0;
} */


#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...