結果

問題 No.1248 Kth Sum
ユーザー chocorusk
提出日時 2020-12-28 10:58:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,412 bytes
コンパイル時間 1,300 ms
コンパイル使用メモリ 128,184 KB
最終ジャッジ日時 2025-01-17 07:52:21
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18 TLE * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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=1e18;
    for(int i=k-1; i<n; i++){
        int m=(i+k-2)/(k-1);
        if(m==1){
            ans=min(ans, a[i]);
            continue;
        }
        priority_queue<ll, vector<ll>, greater<ll>> que;
        for(int j=n-1; j>=k*m-1; j--){
            que.push(a[j]);
        }
        if(que.empty()) continue;
        ll ans1=a[i];
        ans1+=que.top(); que.pop();
        for(int j=m-1; j>=2; j--){
            for(int l=max(i+1, k*j-1); l<k*(j+1)-1; l++){
                que.push(a[l]);
            }
            if(que.empty()){
                ans1=1e18;
                break;
            }
            ans1+=que.top(); que.pop();
        }
        ans=min(ans, ans1);
    }
    cout<<ans<<endl;
    return 0;
}
0