結果

問題 No.615 集合に分けよう
ユーザー pinpin
提出日時 2017-12-15 13:21:50
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,438 bytes
コンパイル時間 840 ms
コンパイル使用メモリ 92,140 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-05-08 12:12:52
合計ジャッジ時間 2,100 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 WA -
testcase_04 AC 2 ms
6,944 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 2 ms
6,944 KB
testcase_08 WA -
testcase_09 AC 2 ms
6,940 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 9 ms
6,944 KB
testcase_16 WA -
testcase_17 AC 34 ms
6,940 KB
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 AC 2 ms
6,944 KB
testcase_26 AC 2 ms
6,940 KB
testcase_27 AC 14 ms
6,940 KB
testcase_28 AC 13 ms
6,940 KB
testcase_29 AC 30 ms
6,944 KB
testcase_30 AC 29 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <cstring>
#include <algorithm> 
#include <math.h>
#include <queue>
#include <functional>
#include <map>
#include <vector>
#include <string>
using namespace std;
typedef long long ll;
const ll MOD = 1000000007;

int n, k;
ll a[100005];
vector<pair<ll, int> > pairs;
int main(void){
    cin >> n >> k;
    for (int i = 0; i < n; i++) cin >> a[i];
    sort(a, a + n);

    for (int i = 0; i < n; i++) {
        if (i == 0){
            pairs.push_back({ a[i + 1] - a[i], i });
        }
        else if (i == n - 1){
            pairs.push_back({ a[i] - a[i - 1], i });
        }
        else{
            pairs.push_back({ a[i + 1] - a[i - 1], i });
        }
    }

    sort(pairs.begin(), pairs.end());

    int cnt = 1;
    bool one[100005];
    for (int i = 0; i < n; i++) one[i] = false;
    one[-1] = one[n] = true;

    for (int i = n - 1; i >= 0; i--){
        int s = pairs[i].second;
        if (one[s - 1] || one[s + 1]){
            one[s] = true;
            cnt++;
        }
        else if (cnt < k - 1){
            one[s] = true;
            cnt += 2;
        }
        if (cnt == k) break;
    }

    ll ans = 0;

    int pos = 0, st;
    while (pos < n){
        st = pos;
        while (!one[pos]){
            pos++;
        }
        if (pos == st){
            pos++;
            continue;
        }
        ans += a[pos - 1] - a[st];
    }

    cout << ans << endl;

}
0