結果
問題 |
No.1248 Kth Sum
|
ユーザー |
![]() |
提出日時 | 2020-12-28 14:00:16 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 325 ms / 2,000 ms |
コード長 | 1,806 bytes |
コンパイル時間 | 1,304 ms |
コンパイル使用メモリ | 135,296 KB |
最終ジャッジ日時 | 2025-01-17 07:53:23 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <stack> #define popcount __builtin_popcount using namespace std; typedef long long int ll; typedef pair<int, int> P; int main() { int n, k; cin>>n>>k; ll a[200020]; for(int i=0; i<n; i++){ cin>>a[i]; } if(k==1){ cout<<a[0]<<endl; return 0; } ll ans=a[k-1]; priority_queue<P, vector<P>, greater<P>> que; priority_queue<P> que1; ll s=0; bool dame[200020]={}; for(int j=n/k; j>=2; j--){ for(int i=k*j-1; i<min(n, k*(j+1)-1); i++){ que.push({a[i], i}); } que1.push(que.top()); s+=que.top().first; dame[que.top().second]=1; que.pop(); } multiset<ll> st, st2, sti; for(int i=k-1; i<=(k-1)*(n/k); i++) if(!dame[i]) st.insert(a[i]); for(int i=k-1; i<n; i++) if(!dame[i]) st2.insert(a[i]); for(int i=0; i<n; i++) if(dame[i]) sti.insert(i); for(int i=n/k; i>=2; i--){ if(*sti.begin()<=(k-1)*i) ans=min(ans, *st2.begin()+s); else ans=min(ans, *st.begin()+s); s-=que1.top().first; int t=que1.top().second; dame[t]=0; sti.erase(t); if(t<=(k-1)*i){ st.insert(a[t]); } st2.insert(a[t]); que1.pop(); for(int j=(k-1)*i; j>(k-1)*(i-1); j--){ if(!dame[j]){ st.erase(st.lower_bound(a[j])); } } } cout<<ans<<endl; return 0; }