結果
問題 |
No.1248 Kth Sum
|
ユーザー |
![]() |
提出日時 | 2020-08-09 00:09:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 128 ms / 2,000 ms |
コード長 | 2,226 bytes |
コンパイル時間 | 2,657 ms |
コンパイル使用メモリ | 201,232 KB |
最終ジャッジ日時 | 2025-01-12 19:09:10 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
#include <bits/stdc++.h> /*#include <boost/multiprecision/cpp_dec_float.hpp> #include <boost/multiprecision/cpp_int.hpp> */#pragma GCC target("avx") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") using namespace std; /*namespace mp = boost::multiprecision; using Bint = mp::cpp_int; using Real = mp::number<mp::cpp_dec_float<1024>>; */#define lli long long int #define uli unsigned long long int #define INF 9999999999999999 #define rep(i,m,n) for(lli i = m;i < n;i++) #define rrep(i,m,n) for(lli i=m-1;i>=n;i--) #define pb(n) push_back(n) #define UE(N) N.erase(unique(N.begin(),N.end()),N.end()); #define Sort(n) sort(n.begin(), n.end()) #define Rev(n) reverse(n.begin(),n.end()) #define Out(S) cout << S << endl #define NeOut(S) cout << S #define HpOut(S) cout << setprecision(50) << S << endl #define Vec(K,L,N,S) vector<L> K(N,S) #define DV(K,L,N,M,S) vector<vector<L>> K(N,vector<L>(M,S)) #define TV(K,L,N,M,R,S) vector<vector<vector<L>>> K(N,vector<vector<L>>(M,vector<L>(R,S))) #define pint pair<lli,lli> #define paf(L,R) pair<L,R> #define mod 1000000007 #define MAX 10000010 #define ALL(a) a.begin(),a.end() #define chmax(a, b) a = (((a)<(b)) ? (b) : (a)) #define chmin(a, b) a = (((a)>(b)) ? (b) : (a)) int main() { long long N, K, sum = 0, num = INF; cin >> N >> K; vector<long long> A(N); vector<long long> CM(N, INF); //後ろからA[i]の累積min rep(i, 0, N) cin >> A[i]; if(!(K - 1)) cout << A[0] << endl; //K=1なら何もしないのが最適 else { CM[N - 1] = A[N - 1]; rrep(i, N-1, 0) CM[i] = min(CM[i + 1], A[i]); priority_queue<long long> pque; //max{A[i+1,min(N,i+X)]} rrep(i, N, K - 1) { //前から見て最初にK番目になるindex long long X = (i + K - 2) / (K - 1); //前から見てi番目が最初にK番目になるとき,部分列の個数の最小値 if(!(X - 1)) chmin(num, A[i]); //部分列の個数が1つならこの時点で確定 else if(K * X <= N) { while(pque.size() - X + 1) { sum -= pque.top(); pque.pop(); } if(CM[K * X - 1] <= pque.top()) chmin(num, A[i] + sum); else chmin(num, A[i] + sum + CM[K * X - 1] - pque.top()); } pque.push(A[i]); sum += A[i]; } cout << num << endl; } }