#977607 (C++23) No.615 集合に分けよう

提出ソース

結果

問題	No.615 集合に分けよう
ユーザー	みしあ
提出日時	2024-04-27 10:14:34
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	33 ms / 2,000 ms
コード長	8,540 bytes
コンパイル時間	7,788 ms
コンパイル使用メモリ	336,108 KB
実行使用メモリ	6,944 KB
最終ジャッジ日時	2024-04-27 10:14:45
合計ジャッジ時間	9,646 ms
ジャッジサーバーID （参考情報）	judge3 / judge5

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	2 ms 6,944 KB
testcase_04	AC	2 ms 6,940 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,940 KB
testcase_07	AC	1 ms 6,940 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	1 ms 6,940 KB
testcase_10	AC	1 ms 6,944 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,944 KB
testcase_13	AC	2 ms 6,944 KB
testcase_14	AC	2 ms 6,940 KB
testcase_15	AC	8 ms 6,944 KB
testcase_16	AC	29 ms 6,940 KB
testcase_17	AC	33 ms 6,944 KB
testcase_18	AC	32 ms 6,944 KB
testcase_19	AC	29 ms 6,940 KB
testcase_20	AC	30 ms 6,940 KB
testcase_21	AC	30 ms 6,940 KB
testcase_22	AC	28 ms 6,940 KB
testcase_23	AC	11 ms 6,940 KB
testcase_24	AC	26 ms 6,940 KB
testcase_25	AC	2 ms 6,944 KB
testcase_26	AC	2 ms 6,940 KB
testcase_27	AC	9 ms 6,940 KB
testcase_28	AC	9 ms 6,944 KB
testcase_29	AC	25 ms 6,940 KB
testcase_30	AC	24 ms 6,940 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#pragma region Macros
#if defined(ONLINE_JUDGE)
#include <atcoder/all>
#endif
#if defined(LOCAL) || defined(_DEBUG)
#include "template.hpp"
#else
#include <bits/stdc++.h>
#include <boost/format.hpp>
using namespace std;
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#define O(...)
#define OP(...)
#define START()
#define STOP()
#define MEMORY()
#define massert(x, msec) if(!(x)) {\
    while(msec > chrono::duration_cast<chrono::milliseconds>(chrono::system_clock::now() - _MY_START_TIME).count()); \
    cerr << "assert [" << __FUNCTION__ << ":" << to_string(__LINE__) << "] : " << string(#x) << el; exit(1);\
}
/******************************************* マクロ等 **************************************************/
#define REP(i, n) for(ll i=0, i##_len=(n); i<i##_len; ++i)
#define REPR(i, n) for(ll i=(n); i>=0; --i)
#define FOR(i, n, m) for(ll i=(m), i##_len=(n); i<i##_len; ++i)
#define EACH(i, v) for(auto&& i : v)
#define ALL(x) (x).begin(),(x).end()
#define ALLR(x) (x).rbegin(),(x).rend()
#define FILL(v, h) memset((v), h, sizeof(v))
#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end())
vector<long long> __FREP_CNT_ARRAY;
#define FREP(i, n) ;__FREP_CNT_ARRAY.push_back(n); REP(i, n) {
#define FREPR(i, n) ;__FREP_CNT_ARRAY.push_back((n)+1); REPR(i, n) {
#define FFOR(i, n, m) ;__FREP_CNT_ARRAY.push_back((n)-(m)); FOR(i, n, m) {
#define FEACH(a, as) ;__FREP_CNT_ARRAY.push_back((as).size()); EACH(a, as) {
#define THEN __FREP_CNT_ARRAY.back()--;} if(bool frep_tmp = !__FREP_CNT_ARRAY.back(); __FREP_CNT_ARRAY.pop_back(), frep_tmp)
#define DPINIT(name) name; bool __SEEN_##name;
#define DP(name) name;if(__SEEN_##name)return name;__SEEN_##name=true;
template<class T, class U>bool chmax(T &a, const U &b) { if (a<(T)b) { a=(T)b; return 1; } return 0; }
template<class T, class U>bool chmin(T &a, const U &b) { if (b<(T)a) { a=(T)b; return 1; } return 0; }
#define cmp3(a, x, b) ((a) <= (x) && (x) < (b))
#define vec vector
#define umap unordered_map
#define uset unordered_set
using ll = long long;
using ull = unsigned long long;
using ld = long double;
using P = pair<ll, ll>;
using Tup = tuple<ll, ll, ll>;
using vl = vec<ll>;
using vvl = vec<vl>;
#define fi first
#define se second
#define pb push_back
#define el '\n'
constexpr ll INF = numeric_limits<ll>::max()/4-1;
/******************************************* 入力 **************************************************/
template<class T> istream &operator>>(istream &stream, vec<T>& o){REP(i, o.size())stream >> o[i];return stream;}
template<class T, class U> istream &operator>>(istream &stream, pair<T, U>& o){cin >> o.fi >> o.se; return stream;}
namespace myinput {
    void input() {}
    template<class T, class... Ts> void input(T&& o, Ts&&... args){cin >> o;input(forward<Ts>(args)...);}
    void assign_vl(size_t) {};
    template<class T, class... Ts> void assign_vl(size_t siz, T&& o, Ts&&... args){o.resize(siz);assign_vl(siz, forward<Ts>(args)...);}
    void input_horizon_sub(size_t) {};
    template<class T, class... Ts> void input_horizon_sub(size_t index, T&& o, Ts&&... args) {cin>>o[index];input_horizon_sub(index, forward<Ts>(args)...);}
    template<class... Ts> void input_horizon(size_t siz, Ts&&... args){REP(i, siz) input_horizon_sub(i, forward<Ts>(args)...);}
}
#define _I(T, ...) ;myinput::input(__VA_ARGS__);
#define I(T, ...) ;T __VA_ARGS__;_I(T, __VA_ARGS__);
#define _Iv(T, siz, ...) ;myinput::assign_vl(siz, __VA_ARGS__);myinput::input(__VA_ARGS__);
#define Iv(T, siz, ...) ;vec<T> __VA_ARGS__;_Iv(T, siz, __VA_ARGS__);
#define _Ih(T, siz, ...) ;myinput::assign_vl(siz, __VA_ARGS__);myinput::input_horizon(siz, __VA_ARGS__);
#define Ih(T, siz, ...) ;vec<T> __VA_ARGS__;_Ih(T, siz, __VA_ARGS__);
chrono::system_clock::time_point _MY_START_TIME;
/******************************************* 出力 **************************************************/
namespace __PrintHelper {
    template <char Separator = ' ', typename T>
    void printValue(std::ostream& os, const T& value) { os << value; }
    template <char Separator = ' ', typename T1, typename T2>
    void printValue(std::ostream& os, const std::pair<T1, T2>& value) { os << value.first << Separator << value.second; }
    template <char Separator = ' ', typename... Args, std::size_t... Is>
    void printTuple(std::ostream& os, const std::tuple<Args...>& value, std::index_sequence<Is...>) { ((os << (Is == 0 ? "" : string(1, Separator)) << std::get<Is>(value)), ...); }
    template <char Separator = ' ', typename... Args>
    void printValue(std::ostream& os, const std::tuple<Args...>& value) { printTuple<Separator>(os, value, std::index_sequence_for<Args...>{}); }
    template <char Separator = ' ', typename T>
    void printValue(std::ostream& os, const std::vector<T>& value) { for (std::size_t i = 0; i < value.size(); ++i) os << (i == 0 ? "" : string(1, Separator)) << value[i]; }
};
template <char Separator = ' ', typename Arg, typename... Args>
void OO(const Arg &x, const Args&... args) {
    __PrintHelper::printValue<Separator>(cout, x);
    if constexpr (sizeof...(args) > 0) {
        cout << Separator;
        OO<Separator>(args...);
    }
    else { cout << el; }
}
#define YN_1(b) {cout << ((b) ? "Yes" : "No") << el;}
#define YN_3(b, y, n) {if(b) cout << (y) << el; else cout << (n) << el;}
#define GET_MACRO(_1,_2,_3,NAME,...) NAME
#define YN(...) GET_MACRO(__VA_ARGS__, YN_3, _, YN_1)(__VA_ARGS__)
#define YNc(d, i, n) YN((d).contains(i), d.at(i), n);
/******************************************* 配列操作 **************************************************/
namespace __ArrayHelper {
    template <typename... Args>
    auto flatten(const std::tuple<Args...>& t);
    template <typename T, std::size_t... Indices>
    auto flattenHelper(const T& t, std::index_sequence<Indices...>) { return std::make_tuple(std::get<Indices>(t)...); }
    template <typename T, typename U>
    auto flatten(const std::pair<T, U>& p) { return flatten(std::make_tuple(p.first, p.second)); }
    template <typename T>
    auto flatten(const T& value) { return std::make_tuple(value); }
    template <typename... Args, std::size_t... Indices>
    auto flattenTupleHelper(const std::tuple<Args...>& t, std::index_sequence<Indices...>) { return std::tuple_cat(flatten(std::get<Indices>(t))...); }
    template <typename... Args>
    auto flatten(const std::tuple<Args...>& t) { return flattenTupleHelper(t, std::index_sequence_for<Args...>{}); }
    template <typename Container>
    auto __flatten(const Container& container) {
        using ValueType = typename Container::value_type;
        using ResultType = decltype(__ArrayHelper::flatten(std::declval<ValueType>()));
        std::vector<ResultType> result;
        for (const auto& item : container) { result.emplace_back(__ArrayHelper::flatten(item)); }
        return result;
    }
    template <class T>
    auto __enumrate(const vec<T> &v) {
        vec<pair<ll, T>> ret(v.size());
        REP(i, v.size()) ret[i] = {i, v[i]};
        return ret;
    }
}
constexpr auto flatten = [](auto&& x) { return __ArrayHelper::__flatten(x); };
constexpr auto enumrate = [](auto&& x) { return __ArrayHelper::__enumrate(x); };
template <typename... Args>
auto zip(const Args&... args) {
    size_t minSize = std::min({args.size()...});
    std::vector<std::tuple<typename Args::value_type...>> result;
    for (size_t i = 0; i < minSize; ++i) { result.emplace_back(std::make_tuple(args[i]...)); }
    return result;
}
template <typename Arg>
Arg compose(Arg arg) { return arg; }
template <typename Arg, typename Func>
auto compose(Arg arg, Func func) { return func(arg); }
template <typename Arg, typename Func, typename... Funcs>
auto compose(Arg arg, Func func, Funcs... funcs) { return compose(compose(arg, func), funcs...); }
/******************************************* Main **************************************************/
#endif
#pragma endregion
using Point = complex<ll>;
void Main();

int main(){
    _MY_START_TIME = std::chrono::system_clock::now();
    std::cin.tie(nullptr);
    std::cout << std::fixed << std::setprecision(15);
    std::cerr << std::fixed << std::setprecision(15);
    Main();
    MEMORY();
    return 0;
}

void Main() {
    I(ll, n, k);
    Iv(ll, n, as);
    sort(ALL(as));
    ll ans = as.back() - as.front();
    O(ans);
    O(as);
    vl diff;
    REP(i, n-1) {
        diff.pb(as[i+1]-as[i]);
    }
    O(diff);
    sort(ALLR(diff));
    O(diff);
    REP(i, min(k-1, n-1)) {
        ans -= diff[i];
    }
    OO(ans);
}

yukicoder

結果

テストケース

ソースコード