Submission #498909

#	Time	Username	Problem	Language	Result	Execution time	Memory
498909		aSSSd	Stove (JOI18_stove)	C++14	100 / 100	74 ms	2292 KiB

stove

#pragma GCC optimize("O2,unroll-loops")
#pragma GCC target("avx,avx2")
#include <bits/stdc++.h>
using namespace std;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int rnd(int l,int r)
{
    return l+rng()%(r-l+1);
}
#define fasty ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define forinc(x,a,b) for(int x=a;x<=b;x++)
#define fordec(x,a,b) for(int x=a;x>=b;x--)
#define forv(a,b) for(auto&a:b)
#define fi first
#define se second
#define pb push_back
#define ii pair<int,int>
#define getbit(x,i) ((x>>(i))&1)
#define batbit(x,i) (x|(1ll<<(i)))
#define tatbit(x,i) (x&~(1<<(i)))
#define all(a) a.begin() , a.end()
int n,k;
int a[1000000+100];


main()
{
    int t;
    t=1;
    //cin >> t;
    while(t--)
    {
        priority_queue<int, vector<int> > h;
        cin >> n >> k;
        forinc(i,1,n) cin >> a[i];
        long long  ans = a[n] - a[1] + 1;
        forinc(i,1,n-1) h.push(a[i+1] - a[i]);
        forinc(i,1,k-1)
        {
            ans -= h.top() ;
            ans+=1;
            h.pop();
        }
        cout << ans << "\n";
    }
}

Compilation message (stderr)

stove.cpp:26:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   26 | main()
      | ^~~~

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