Submission #268183

#	Time	Username	Problem	Language	Result	Execution time	Memory
268183		leductoan	Stove (JOI18_stove)	C++14	100 / 100	23 ms	3300 KiB

stove

#include<bits/stdc++.h>
using namespace std;
#define task "stove"
#define lb lower_bound
#define ub upper_bound
#define alll(v) (v).begin(),(v).end()
#define szz(v) (v).size()
#define fi first
#define se second
#define pb push_back
#define FOR(i,a,b) for (ll i = (a), _b = (b); i <= _b; ++i)
#define FORD(i,b,a) for (ll i = (b), _a = (a); i >= _a; --i)
typedef long double ld;
typedef long long ll;
typedef pair<ll,ll> pii;

const ll MOD=1000000007; /// 998244353
const ll base=311; /// for hash actually
const ll N=1e5+5;

ll n,k,a[N];

void solved()
{
    cin>>n>>k;
    --k;
    FOR(i,1,n) cin>>a[i];
    ll ans=a[n]-a[1]+1;
    vector<ll> v;
    FOR(i,2,n) v.pb(a[i]-a[i-1]-1);
    sort(alll(v),greater<ll>());
    FOR(i,0,k-1)
    {
        ans-=v[i];
    }
    cout<<ans;
}
int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0);
    if(fopen(task".inp","r"))
    {
        freopen(task".inp","r",stdin);
        freopen(task".out","w",stdout);
    }
    solved();
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:43:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   43 |         freopen(task".inp","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
stove.cpp:44:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
   44 |         freopen(task".out","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~

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