結果
問題 | 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]の累積minrep(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番目になるindexlong 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;}}