結果

問題 No.1248 Kth Sum
ユーザー 👑 tute7627tute7627
提出日時 2020-07-19 15:32:12
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 50 ms / 2,000 ms
コード長 1,410 bytes
コンパイル時間 2,007 ms
コンパイル使用メモリ 205,052 KB
実行使用メモリ 5,332 KB
最終ジャッジ日時 2023-08-22 10:00:56
合計ジャッジ時間 7,251 ms
ジャッジサーバーID
(参考情報)
judge13 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 2 ms
4,376 KB
testcase_09 AC 1 ms
4,380 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 2 ms
4,376 KB
testcase_12 AC 1 ms
4,384 KB
testcase_13 AC 27 ms
5,284 KB
testcase_14 AC 27 ms
4,636 KB
testcase_15 AC 27 ms
5,184 KB
testcase_16 AC 37 ms
4,916 KB
testcase_17 AC 33 ms
4,688 KB
testcase_18 AC 31 ms
4,600 KB
testcase_19 AC 31 ms
4,608 KB
testcase_20 AC 29 ms
4,532 KB
testcase_21 AC 27 ms
4,764 KB
testcase_22 AC 27 ms
4,548 KB
testcase_23 AC 2 ms
4,376 KB
testcase_24 AC 41 ms
5,332 KB
testcase_25 AC 50 ms
5,200 KB
testcase_26 AC 44 ms
5,236 KB
testcase_27 AC 24 ms
4,592 KB
testcase_28 AC 24 ms
4,592 KB
testcase_29 AC 23 ms
4,380 KB
testcase_30 AC 37 ms
4,944 KB
testcase_31 AC 32 ms
4,600 KB
testcase_32 AC 31 ms
4,600 KB
testcase_33 AC 1 ms
4,376 KB
testcase_34 AC 40 ms
4,880 KB
testcase_35 AC 32 ms
4,632 KB
testcase_36 AC 30 ms
4,592 KB
testcase_37 AC 28 ms
4,548 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;

#define ALL(a)  (a).begin(),(a).end()
#define ALLR(a)  (a).rbegin(),(a).rend()
#define rep(i,n,m) for(int i = (n); i < (int)(m); i++)
#define rrep(i,n,m) for(int i = (m) - 1; i >= (int)(n); i--)
using ll = long long;
using ld = long double;
const ll MOD1 = 1e9+7;
const ll MOD9 = 998244353;
const int INF = (int)1e9+1;
template<typename T>
bool chmin(T &a,T b){if(a>b){a=b;return true;}else return false;}
template<typename T>
bool chmax(T &a,T b){if(a<b){a=b;return true;}else return false;}
template<typename T>
vector<ll>dx={1,0,-1,0,1,1,-1,-1};
vector<ll>dy={0,1,0,-1,1,-1,1,-1};
int main(){
  cin.tie(nullptr);
  ios_base::sync_with_stdio(false);
  int n,k;
  cin>>n>>k;
  assert(1<=n&&n<=200000);
  assert(1<=k&&k<=n);
  vector<int>a(n);
  rep(i,0,n){
    cin>>a[i];
    assert(1<=a[i]&&a[i]<=(int)1e9);
  }
  if(k==1){
    cout<<a[0]<<endl;
    return 0;
  }
  priority_queue<int>que;
  ll sum=0;
  ll res=INF;
  vector<int>b(n,INF);
  b[n-1]=a[n-1];
  rrep(i,0,n-1)b[i]=min(b[i+1],a[i]);
  rrep(i,k-1,n){
    int m=(i+k-2)/(k-1);
    if(m==1){
      chmin(res,(ll)a[i]);
    }
    else if(m*k<=n){
      while(que.size()>m-1){
        sum-=que.top();
        que.pop();
      }
      if(b[m*k-1]<=que.top())chmin(res,a[i]+sum);
      else chmin(res,a[i]+sum-que.top()+b[m*k-1]);
    }
    que.push(a[i]);
    sum+=a[i];
  }
  cout<<res<<endl;
  return 0;
}
0