Submission #1287638

#	Time	Username	Problem	Language	Result	Execution time	Memory
1287638		maxphast	Stove (JOI18_stove)	C++20	100 / 100	18 ms	2788 KiB

stove

/*
* @Author: MaxPhast
* @File:   test.cpp
* @Date:   2025-11-05 03:41:37
*/
#include <bits/stdc++.h>
using namespace std;

#define int long long
#define ii pair <int , int>
#define iii pair <int , ii>
#define iiii pair <ii , ii>
#define FOR(i , l , r) for(int i = (l) , _r = (r) ; i <= _r ; ++ i)
#define FORD(i , r , l) for(int i = (r) , _l = (l) ; i >= _l ; -- i)
#define ALL(vec) vec.begin() , vec.end()
#define UNI(vec) sort(ALL(vec)) , vec.erase(unique(ALL(vec)) , vec.end())
#define pb push_back
#define MASK(i) (1ll << i)
#define BIT(mask , i) ((mask >> i) & 1ll)
#define ON(mask , i) (mask | MASK(i))
#define OFF(mask , i) (mask & (~MASK(i)))
#define TURN(mask , i) (mask ^ MASK(i))
#define BP(mask) __builtin_popcountll(mask)
#define sqr(x) (1ll * (x) * (x))
#define fi first
#define se second
#define oo 1e18
#define time() cerr << " \n " << "Time : " << 1000.0 * clock() / CLOCKS_PER_SEC << "ms."
#define IO(TASK) if(fopen(#TASK".INP" , "r")){freopen(#TASK".INP" , "r" , stdin); freopen(#TASK".OUT" , "w" , stdout);}

template <class X , class Y>
    bool maxz(X &a , const Y b)
    {
        if(a < b)
        {
            a = b;
            return true;
        }
        return false;
    }
template <class X , class Y>
    bool minz(X &a , const Y b)
    {
        if(a > b)
        {
            a = b;
            return true;
        }
        return false;
    }

namespace MaxPhast
{

    const int N = 1e5 + 5;

    int n , k;
    int a[N];
    ii dp[N];

    ii f(int lambda)
    {
        dp[1] = {lambda + 1 , 0};
        FOR(i , 2 , n)
        {
            dp[i] = min(ii(dp[i - 1].fi + a[i] - a[i - 1] , dp[i - 1].se) , 
                        ii(dp[i - 1].fi + 1 + lambda , dp[i - 1].se - 1));
        }
        return dp[n];
    }

    void solve()
    {
        cin >> n >> k;

        FOR(i , 1 , n)
            cin >> a[i];

        sort(a + 1 , a + 1 + n);

        int l = 0 , r = 1e9 , res = 0;
        while(l <= r)
        {
            int mid = l + r >> 1;
            if(-f(mid).se >= k)
            {
                res = mid;
                l = mid + 1;
            }
            else
                r = mid - 1;
        }

        cout << f(res).fi - k * res;
    }

}

signed main()
{
    ios_base :: sync_with_stdio(false) ; cin.tie(nullptr) ; cout.tie(nullptr);
    IO(test);
    MaxPhast :: solve();
    time();
}
/*

*/

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:29:54: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   29 | #define IO(TASK) if(fopen(#TASK".INP" , "r")){freopen(#TASK".INP" , "r" , stdin); freopen(#TASK".OUT" , "w" , stdout);}
      |                                               ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~
stove.cpp:102:5: note: in expansion of macro 'IO'
  102 |     IO(test);
      |     ^~
stove.cpp:29:90: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   29 | #define IO(TASK) if(fopen(#TASK".INP" , "r")){freopen(#TASK".INP" , "r" , stdin); freopen(#TASK".OUT" , "w" , stdout);}
      |                                                                                   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
stove.cpp:102:5: note: in expansion of macro 'IO'
  102 |     IO(test);
      |     ^~

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