Submission #659523

#	Time	Username	Problem	Language	Result	Execution time	Memory
659523		onebit1024	Stove (JOI18_stove)	C++17	0 / 100	1 ms	320 KiB

stove

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define pb push_back
#define all(c) c.begin(), c.end()
#define endl "\n"

const double PI=3.141592653589;


void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");}

template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? "," : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#ifndef ONLINE_JUDGE
#define dbg(x...) cerr << "LINE(" << __LINE__ << ") -> " <<"[" << #x << "] = ["; _print(x)
#else
#define dbg(x...)
#endif


vector<int>a;
int n,k;
pair<bool,int>f(int x){
    int pos = 0;
    int res = 0, cnt = 0;
    for(int i = 0;i<=n;++i){
        if(pos >= n)break;
        int val = lower_bound(all(a), a[pos]+x)-a.begin() - 1;
        cnt++;
        res+=(a[val]+1 - a[pos]);
        pos = val+1;
    }
    if(pos < n)return {0,0};
    return {cnt<=k,res};
}
void solve()
{
    cin >> n >> k;
    a.resize(n);
    for(int &u : a)cin >> u;
    sort(all(a));
    int l = 0, r = 1e10;
    int res = 1;
    while(l+1 < r){
        int m = (l+r)/2;
        pair<int,int>c = f(m);
        if(c.first){
            res = c.second;
            r = m;
        }else l = m;
    }
    cout << res;
}   

int32_t main()
{

    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);


    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);
    

    int T=1;
    for(int i = 1;i<=T;++i)
    {
        // cout << "Case #" << i << ": ";
        solve();
    }
}

• Subtask 1: 0 / 20
• Subtask 2: 0 / 30
• Subtask 3: 0 / 50