結果

問題 No.1248 Kth Sum
ユーザー ThistleThistle
提出日時 2020-10-03 00:42:49
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 8,390 bytes
コンパイル時間 2,326 ms
コンパイル使用メモリ 142,472 KB
実行使用メモリ 46,824 KB
最終ジャッジ日時 2023-09-25 01:18:16
合計ジャッジ時間 8,738 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 17 ms
34,616 KB
testcase_01 AC 17 ms
34,688 KB
testcase_02 AC 17 ms
34,608 KB
testcase_03 AC 16 ms
34,616 KB
testcase_04 AC 17 ms
34,696 KB
testcase_05 AC 16 ms
34,912 KB
testcase_06 AC 17 ms
34,692 KB
testcase_07 AC 17 ms
34,720 KB
testcase_08 AC 17 ms
34,608 KB
testcase_09 AC 18 ms
34,628 KB
testcase_10 AC 17 ms
34,652 KB
testcase_11 AC 17 ms
34,680 KB
testcase_12 AC 17 ms
34,616 KB
testcase_13 AC 118 ms
46,752 KB
testcase_14 AC 159 ms
46,768 KB
testcase_15 AC 110 ms
46,708 KB
testcase_16 AC 162 ms
46,600 KB
testcase_17 AC 169 ms
46,820 KB
testcase_18 AC 169 ms
46,708 KB
testcase_19 AC 167 ms
46,772 KB
testcase_20 AC 165 ms
46,708 KB
testcase_21 AC 165 ms
46,708 KB
testcase_22 AC 149 ms
46,712 KB
testcase_23 AC 17 ms
34,628 KB
testcase_24 AC 117 ms
43,632 KB
testcase_25 AC 167 ms
46,600 KB
testcase_26 AC 139 ms
46,728 KB
testcase_27 AC 90 ms
46,824 KB
testcase_28 AC 89 ms
46,596 KB
testcase_29 AC 89 ms
46,732 KB
testcase_30 AC 102 ms
46,812 KB
testcase_31 AC 104 ms
46,600 KB
testcase_32 AC 104 ms
46,600 KB
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target ("avx")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define _USE_MATH_DEFINES
#include<iostream>
#include<string>
#include<queue>
#include<cmath>
#include<map>
#include<set>
#include<list>
#include<iomanip>
#include<vector>
#include<random>
#include<functional>
#include<algorithm>
#include<stack>
#include<cstdio>
#include<cstring>
#include<bitset>
#include<unordered_map>
#include<climits>
#include<fstream>
#include<complex>
#include<time.h>
#include<cassert>
#include<functional>
#include<numeric>
#include<tuple>
using namespace std;
using ll = long long;
using ld = long double;
using H = pair<ll, ll>;
using P = pair<ll, H>;
using vi = vector<ll>;
#define all(a) (a).begin(),(a).end()
#define fs first
#define sc second
#define xx first
#define yy second.first
#define zz second.second
#define Q(i,j,k) mkp(i,mkp(j,k))
#define rng(i,s,n) for(ll i = (s) ; i < (n) ; i++)
#define rep(i,n) rng(i, 0, (n))
#define mkp make_pair
#define vec vector
#define pb emplace_back
#define siz(a) (int)(a).size()
#define crdcomp(b) sort(all((b)));(b).erase(unique(all((b))),(b).end())
#define getidx(b,i) (lower_bound(all(b),(i))-(b).begin())
#define ssp(i,n) (i==(ll)(n)-1?"\n":" ")
#define ctoi(c) (int)(c-'0')
#define itoc(c) (char)(c+'0')
#define cyes printf("Yes\n")
#define cno printf("No\n")
#define cdf(n) for(int quetimes_=(n);quetimes_>0;quetimes_--)
#define gcj printf("Case #%lld: ",qq123_+1)
#define readv(a,n) a.resize(n,0);rep(i,(n)) a[i]=read()
#define found(a,x) (a.find(x)!=a.end())
constexpr ll mod = (ll)1e9 + 7;
constexpr ll Mod = 998244353;
constexpr ld EPS = 1e-10;
constexpr ll inf = (ll)3 * 1e18;
constexpr int Inf = (ll)15 * 1e8;
constexpr int dx[] = { -1,1,0,0 }, dy[] = { 0,0,-1,1 };
template<class T>bool chmax(T& a, const T& b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; }
ll read() { ll u, k = scanf("%lld", &u); return u; }
string reads() { string s; cin >> s; return s; }
H readh(short g = 0) { H u; int k = scanf("%lld %lld", &u.fs, &u.sc); if (g == 1) u.fs--, u.sc--; if (g == 2) u.fs--; return u; }
bool ina(H t, int h, int w) { return 0 <= t.fs && t.fs < h && 0 <= t.sc && t.sc < w; }
bool ina(int t, int l, int r) { return l <= t && t < r; }
ll gcd(ll i, ll j) { return j ? gcd(j, i % j) : i; }
ll popcount(ll x) {
    int sum = 0; for (int i = 0; i < 60; i++)if ((1ll << i) & x) sum++;
    return sum;
}
template<typename T>
class csum {
    vec<T> v;
public:
    csum(vec<T>& a) :v(a) { build(); }
    csum() {}
    void init(vec<T>& a) { v = a; build(); }
    void build() {
        for (int i = 1; i < v.size(); i++) v[i] += v[i - 1];
    }
    //[l,r]
    T a(int l, int r) {
        if (r < l) return 0;
        return v[r] - (l == 0 ? 0 : v[l - 1]);
    }
    //[l,r)
    T b(int l, int r) {
        return a(l, r - 1);
    }
    T a(pair<int, int>t) {
        return a(t.first, t.second);
    }
    T b(pair<int, int>t) {
        return b(t.first, t.second);
    }
};
class mint {
public:ll v;
      mint(ll v = 0) { s(v % mod + mod); }
      constexpr static int mod = Mod;// (ll)1e9 + 7;
      constexpr static int fn_ = (ll)2e6 + 5;
      static mint fact[fn_], comp[fn_];
      mint pow(int x) const {
          mint b(v), c(1);
          while (x) {
              if (x & 1) c *= b;
              b *= b;
              x >>= 1;
          }
          return c;
      }
      inline mint& s(int vv) {
          v = vv < mod ? vv : vv - mod;
          return *this;
      }
      inline mint inv()const { return pow(mod - 2); }
      inline mint operator-()const { return mint() - *this; }
      inline mint& operator+=(const mint b) { return s(v + b.v); }
      inline mint& operator-=(const mint b) { return s(v + mod - b.v); }
      inline mint& operator*=(const mint b) { v = v * b.v % mod; return *this; }
      inline mint& operator/=(const mint b) { v = v * b.inv().v % mod; return *this; }
      inline mint operator+(const mint b) const { return mint(v) += b; }
      inline mint operator-(const mint b) const { return mint(v) -= b; }
      inline mint operator*(const mint b) const { return mint(v) *= b; }
      inline mint operator/(const mint b) const { return mint(v) /= b; }
      friend ostream& operator<<(ostream& os, const mint& m) {
          return os << m.v;
      }
      friend istream& operator>>(istream& is, mint& m) {
          int x; is >> x; m = mint(x);
          return is;
      }
      bool operator<(const mint& r)const { return v < r.v; }
      bool operator>(const mint& r)const { return v > r.v; }
      bool operator<=(const mint& r)const { return v <= r.v; }
      bool operator>=(const mint& r)const { return v >= r.v; }
      bool operator==(const mint& r)const { return v == r.v; }
      bool operator!=(const mint& r)const { return v != r.v; }
      explicit operator bool()const { return v; }
      explicit operator int()const { return v; }
      mint comb(mint k) {
          if (k > * this) return mint();
          if (!fact[0]) combinit();
          if (v >= fn_) {
              if (k > * this - k) k = *this - k;
              mint tmp(1);
              for (int i = v; i >= v - k.v + 1; i--) tmp *= mint(i);
              return tmp * comp[k.v];
          }
          return fact[v] * comp[k.v] * comp[v - k.v];
      }//nCk
      mint perm(mint k) {
          if (k > * this) return mint();
          if (!fact[0]) combinit();
          if (v >= fn_) {
              mint tmp(1);
              for (int i = v; i >= v - k.v + 1; i--) tmp *= mint(i);
              return tmp;
          }
          return fact[v] * comp[v - k.v];
      }//nPk
      static void combinit() {
          fact[0] = 1;
          for (int i = 1; i < fn_; i++) fact[i] = fact[i - 1] * mint(i);
          comp[fn_ - 1] = fact[fn_ - 1].inv();
          for (int i = fn_ - 2; i >= 0; i--) comp[i] = comp[i + 1] * mint(i + 1);
      }
}; mint mint::fact[fn_], mint::comp[fn_];

//--------------------------------------------------------------


//--------------------------------------------------------------
template<typename T>
class topK {
    int n, rr;
    vector<T>v;
    vector<T>dat;
    vector<int>count;
public:
    topK(vector<T>a) {
        v = a; sort(v.begin(), v.end());
        v.erase(unique(v.begin(), v.end()), v.end());
        n = v.size(); rr = 1;
        while (rr < n) rr <<= 1;
        dat.assign(2 * rr - 1, 0);
        count.assign(2 * rr - 1, 0);
    }
    void add(T x) {
        int pos = lower_bound(v.begin(), v.end(), x) - v.begin();
        pos += rr - 1;
        dat[pos] += x; count[pos]++;
        while (pos > 0) {
            pos = (pos - 1) >> 1;
            dat[pos] = dat[pos * 2 + 1] + dat[pos * 2 + 2];
            count[pos] = count[pos * 2 + 1] + count[pos * 2 + 2];
        }
    }
    T query(T a, T b, int k) {
        int x = lower_bound(v.begin(), v.end(), a) - v.begin();
        int y = lower_bound(v.begin(), v.end(), b) - v.begin();
        return query(0, x, y, 0, rr, k);
    }
private:
    T query(int i, int a, int b, int l, int r, int& k) {
        if (b <= l || r <= a) return 0;
        if (a <= l && r <= b) {
            if (count[i] < k) {
                k -= count[i];
                return dat[i];
            }
            if (r - l == 1) {
                T tmp= dat[i] - (count[i] - k) * v[l];
                k = 0;
                return tmp;
            }
        }
        T sum = query(i * 2 + 1, a, b, l, (l + r) / 2, k);
        if (k > 0) sum += query(i * 2 + 2, a, b, (l + r) / 2, r, k);
        return sum;
    }
};

int n, k;
vi a;
signed main() {
    cin >> n >> k;
    readv(a, n);
    vi b = a; crdcomp(b);
    topK<ll> seg(a);
    ll ans = a[k - 1];
    if (k == 1) {
        cout << a[0] << endl;
        return 0;
    }
    for (int i = n - 1; i > int(n / k) * (k - 1); i--) {
        seg.add(a[i]);
    }
    for (int i = n / k; i >= 2; i--) {
        //k~i*(k-1)で1つ
        //それから後ろでi-1個
        //(i-1)*(k-1)+2~i*(k-1)+1
        for (int j = i * (k - 1); j > (i - 1) * (k - 1); j--) {
            if ((n - j) < (i - 1)) continue;
            ll r = a[j] + seg.query(0, b.back() + 1, i - 1);
            if (r > 0)chmin(ans, r);
            seg.add(a[j]);
        }
    }
    cout << ans << endl;
}
0