結果

問題 No.738 平らな農地
ユーザー ooaiuooaiu
提出日時 2024-12-01 11:34:15
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 856 ms / 2,000 ms
コード長 3,517 bytes
コンパイル時間 3,560 ms
コンパイル使用メモリ 257,272 KB
実行使用メモリ 54,272 KB
最終ジャッジ日時 2024-12-01 11:34:51
合計ジャッジ時間 35,761 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 13 ms
5,248 KB
testcase_06 AC 14 ms
5,248 KB
testcase_07 AC 14 ms
5,248 KB
testcase_08 AC 7 ms
5,248 KB
testcase_09 AC 4 ms
5,248 KB
testcase_10 AC 3 ms
5,248 KB
testcase_11 AC 7 ms
5,248 KB
testcase_12 AC 4 ms
5,248 KB
testcase_13 AC 12 ms
5,248 KB
testcase_14 AC 5 ms
5,248 KB
testcase_15 AC 680 ms
51,584 KB
testcase_16 AC 735 ms
51,968 KB
testcase_17 AC 809 ms
51,840 KB
testcase_18 AC 757 ms
51,328 KB
testcase_19 AC 595 ms
53,632 KB
testcase_20 AC 698 ms
51,968 KB
testcase_21 AC 856 ms
53,632 KB
testcase_22 AC 725 ms
52,224 KB
testcase_23 AC 804 ms
52,608 KB
testcase_24 AC 776 ms
52,224 KB
testcase_25 AC 5 ms
5,248 KB
testcase_26 AC 5 ms
5,248 KB
testcase_27 AC 5 ms
5,248 KB
testcase_28 AC 5 ms
5,248 KB
testcase_29 AC 5 ms
5,248 KB
testcase_30 AC 5 ms
5,248 KB
testcase_31 AC 5 ms
5,248 KB
testcase_32 AC 5 ms
5,248 KB
testcase_33 AC 5 ms
5,248 KB
testcase_34 AC 5 ms
5,248 KB
testcase_35 AC 4 ms
5,248 KB
testcase_36 AC 5 ms
5,248 KB
testcase_37 AC 5 ms
5,248 KB
testcase_38 AC 5 ms
5,248 KB
testcase_39 AC 6 ms
5,248 KB
testcase_40 AC 5 ms
5,248 KB
testcase_41 AC 5 ms
5,248 KB
testcase_42 AC 4 ms
5,248 KB
testcase_43 AC 5 ms
5,248 KB
testcase_44 AC 5 ms
5,248 KB
testcase_45 AC 655 ms
54,144 KB
testcase_46 AC 644 ms
51,456 KB
testcase_47 AC 647 ms
53,120 KB
testcase_48 AC 553 ms
51,968 KB
testcase_49 AC 551 ms
51,712 KB
testcase_50 AC 523 ms
53,120 KB
testcase_51 AC 704 ms
53,248 KB
testcase_52 AC 591 ms
51,840 KB
testcase_53 AC 640 ms
52,352 KB
testcase_54 AC 668 ms
53,504 KB
testcase_55 AC 693 ms
53,376 KB
testcase_56 AC 656 ms
52,864 KB
testcase_57 AC 661 ms
51,328 KB
testcase_58 AC 611 ms
52,736 KB
testcase_59 AC 647 ms
54,272 KB
testcase_60 AC 581 ms
53,760 KB
testcase_61 AC 626 ms
53,632 KB
testcase_62 AC 587 ms
51,328 KB
testcase_63 AC 688 ms
54,144 KB
testcase_64 AC 623 ms
53,632 KB
testcase_65 AC 126 ms
51,968 KB
testcase_66 AC 129 ms
53,120 KB
testcase_67 AC 414 ms
51,584 KB
testcase_68 AC 404 ms
52,480 KB
testcase_69 AC 577 ms
52,864 KB
testcase_70 AC 495 ms
53,888 KB
testcase_71 AC 136 ms
53,504 KB
testcase_72 AC 134 ms
51,456 KB
testcase_73 AC 126 ms
52,096 KB
testcase_74 AC 143 ms
52,864 KB
testcase_75 AC 516 ms
51,328 KB
testcase_76 AC 463 ms
52,480 KB
testcase_77 AC 457 ms
51,456 KB
testcase_78 AC 488 ms
53,760 KB
testcase_79 AC 603 ms
54,016 KB
testcase_80 AC 499 ms
52,736 KB
testcase_81 AC 533 ms
51,712 KB
testcase_82 AC 467 ms
52,992 KB
testcase_83 AC 471 ms
54,272 KB
testcase_84 AC 490 ms
53,120 KB
testcase_85 AC 148 ms
53,888 KB
testcase_86 AC 145 ms
51,328 KB
testcase_87 AC 96 ms
52,736 KB
testcase_88 AC 93 ms
51,328 KB
testcase_89 AC 2 ms
5,248 KB
testcase_90 AC 2 ms
5,248 KB
testcase_91 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#ifndef LOCAL
#include <bits/stdc++.h>

using namespace std;

#define debug(...) (void(0))
#else
#include "algo/debug.h"
#endif
template <class T>
struct MergeSortTree {
   private:
    int N;
    int sz;
    T mn, mx;
    std::vector<std::vector<T>> dat;
    std::vector<std::vector<T>> pref;
    int LB(const std::vector<T>& v, const T& x) const {
        return std::distance(v.begin(), std::lower_bound(v.begin(), v.end(), x));
    }

   public:
    MergeSortTree() = default;
    MergeSortTree(const std::vector<T>& v) : N((int)v.size()) {
        mn = std::numeric_limits<T>::max();
        mx = std::numeric_limits<T>::lowest();
        sz = (int)std::bit_ceil((unsigned int)N);
        dat.assign(2 * sz, {});
        pref.assign(2 * sz, {0});
        for (int i = 0; i < N; i++) {
            dat[i + sz].push_back(v[i]);
            pref[i + sz].push_back(v[i]);
            mn = std::min(mn, v[i]);
            mx = std::max(mx, v[i]);
        }
        for (int i = sz - 1; i >= 1; i--) {
            dat[i].resize(dat[2 * i].size() + dat[2 * i + 1].size());
            std::merge(dat[2 * i].begin(), dat[2 * i].end(), dat[2 * i + 1].begin(), dat[2 * i + 1].end(), dat[i].begin());

            pref[i].resize(dat[i].size() + 1);
            for (unsigned j{}; j < dat[i].size(); j++) {
                pref[i][j + 1] += pref[i][j] + dat[i][j];
            }
        }
    };
    int count_lt(int l, int r, const T& x) const {
        assert(0 <= l && l <= r && r <= N);
        l += sz;
        r += sz;
        int ans = 0;
        while (l < r) {
            if (l & 1) ans += LB(dat[l++], x);
            if (r & 1) ans += LB(dat[--r], x);
            l >>= 1;
            r >>= 1;
        }
        return ans;
    }
    int count_le(int l, int r, const T& x) const { return count_lt(l, r, x + 1); }
    int count_gt(int l, int r, const T& x) const { return r - l - count_le(l, r, x); }
    int count_ge(int l, int r, const T& x) const { return r - l - count_lt(l, r, x); }
    T kth(int l, int r, const T& k) const {
        T lo = mn, hi = mx + T{1};
        while (hi > lo + 1) {
            T mi = (lo + hi) >> 1;
            (count_lt(l, r, mi) <= k ? lo : hi) = mi;
        }
        return lo;
    }
    int64_t Solve(int l, int r, int K) const {
        int64_t val = kth(l, r, K / 2);
        int64_t cnt_lt = count_lt(l, r, val);
        int64_t cnt_ge = r - l - cnt_lt;
        l += sz;
        r += sz;
        int64_t sum_lt = 0;
        int64_t sum_all = 0;
        while (l < r) {
            if (l & 1) {
                int k = LB(dat[l], val);
                sum_lt += pref[l][k];
                sum_all += pref[l][dat[l].size()];
                l++;
            }
            if (r & 1) {
                r--;
                int k = LB(dat[r], val);
                sum_lt += pref[r][k];
                sum_all += pref[r][dat[r].size()];
            }
            l >>= 1;
            r >>= 1;
        }
        int64_t sum_ge = sum_all - sum_lt;
        return val * (cnt_lt - cnt_ge) + sum_ge - sum_lt;
    }
};
void solve() {
    int N, K;
    cin >> N >> K;
    vector<int64_t> A(N);
    for (int i = 0; i < N; i++) cin >> A[i];
    MergeSortTree<int64_t> freq(A);
    int64_t ans = 1e18;
    for (int i = 0; i + K <= N; i++) {
        ans = min(ans, freq.Solve(i, i + K, K));
    }
    cout << ans << endl;
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int tt = 1;
    // std::cin >> tt;
    while (tt--) {
        solve();
    }
}
0