Submission #75468872

---

Source Code Expand

#line 2 "/Users/blue_jam/ProconLibrary/misc/template.hpp"

#include <atcoder/all>
#include <bits/stdc++.h>

using namespace std;
using namespace atcoder;

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

using ll = long long;
#ifndef EPS
const double eps = 1e-10;
#else
const double eps = EPS;
#endif
#line 2 "abc456/F/main.cpp"


// K = 2
// A = 1 1000 1
const ll INF = 1e18;

using S = vector<vector<ll>>;

S op(S a, S b) {
    ll n = a.size(), m = b[0].size();
    S c(n, vector<ll>(m, INF));
    for (ll i = 0; i < n; i++) {
        for (ll j = 0; j < m; j++) {
            for (ll k = 0; k < a[i].size(); k++) {
                c[i][j] = min(c[i][j], a[i][k] + b[k][j]);
            }
        }
    }
    return c;
}

S e() {
    vector r(2, vector<ll>(2, INF));
    r[0][0] = r[1][1] = 0;
    return r;
}

int main() {
    ll T;
    cin >> T;
    for (; T--;) {
        ll N, K;
        cin >> N >> K;
        vector<ll> A(N);
        vector<S> B(N, vector(2, vector<ll>(2)));
        for (ll i = 0; i < N; i++) {
            cin >> A[i];
            B[i][0][0] = A[i];
            B[i][0][1] = A[i];
            B[i][1][0] = 0;
            B[i][1][1] = INF;
        }
        segtree<S, op, e> seg(B);
        ll ans = INF;
        for (ll k = K; k <= K + 1; k++) {
            for (ll i = 0; i + k <= N; i++) {
                S t = seg.prod(i, i + k - 1);
                S u(2, vector<ll>(1));
                u[0][0] = A[i + k - 1];
                u[1][0] = INF;
                S v = op(t, u);
                ans = min(ans, v[0][0]);
            }
        }
        cout << ans << endl;
    }
    return 0;
}

Submission Info

Compile Error

abc456/F/main.cpp: In function 'S op(S, S)':
abc456/F/main.cpp:15:30: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]

Judge Result