結果
問題 | No.615 集合に分けよう |
ユーザー |
|
提出日時 | 2025-03-28 07:50:07 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 22 ms / 2,000 ms |
コード長 | 1,328 bytes |
コンパイル時間 | 1,381 ms |
コンパイル使用メモリ | 95,436 KB |
実行使用メモリ | 7,324 KB |
最終ジャッジ日時 | 2025-03-28 07:50:11 |
合計ジャッジ時間 | 3,973 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 26 |
ソースコード
#include<iostream>#include<vector>#include<algorithm>#include<iomanip>using namespace std;#define OVERLOAD_REP(_1, _2, _3, name, ...) name#define REP1(i, n) for (auto i = std::decay_t<decltype(n)>{}; (i) != (n); ++(i))#define REP2(i, l, r) for (auto i = (l); (i) != (r); ++(i))#define rep(...) OVERLOAD_REP(__VA_ARGS__, REP2, REP1)(__VA_ARGS__)#define REP(i, l, r) rep(i, l, r+1)#define all(x) (x).begin(),(x).end()#define inf 1000000000using ll = long long;using P = pair<int,int>;using Graph = vector<vector<int> >;int main() {// 高速化ios::sync_with_stdio(false);cin.tie(nullptr);// 小数点の出力桁数を指定cout << fixed << setprecision(10);// メインint N, K;cin >> N >> K;vector<ll> a(N);rep(i, 0, N) cin >> a[i];sort(all(a));vector<pair<ll, int> > sa(N-1);rep(i, 0, N-1) sa[i] = {a[i+1]-a[i], i+1};sort(sa.rbegin(), sa.rend());vector<int> wake;wake.push_back(0);rep(i, 0, K-1) wake.push_back(sa[i].second);wake.push_back(N);sort(all(wake));//rep(i, 0, N) cout << a[i] << " ";//cout << endl;//rep(i, 0, K+1) cout << wake[i] << " ";//cout << endl;ll ans = 0;rep(i, 0, K) {ans += a[wake[i+1]-1] - a[wake[i]];}cout << ans << endl;return 0;}