Submission #887808

#	Time	Username	Problem	Language	Result	Execution time	Memory
887808		thunopro	Stove (JOI18_stove)	C++14	100 / 100	16 ms	2344 KiB

stove

#include<bits/stdc++.h>
using namespace std ;
#define maxn 200009 
#define ll long long
#define fi first 
#define se second 
#define pb push_back 
#define left id<<1
#define right id<<1|1 
#define re exit(0); 

const int mod = 1e9+7 ; 
const int INF = 1e9 ; 
const int LOG = 18 ; 

typedef vector<int> vi ; 
typedef pair<int,int> pii ; 
typedef vector<pii> vii ; 
typedef pair<ll,ll> pll ;  

void add ( int &a , int b ) 
{
	a += b ; 
	if ( a < 0 ) a += mod ; 
	if ( a >= mod ) a -= mod ; 
}

template < typename T > void chkmin (int &a , int b) { if (a>b) a=b ;} 
template < typename T > void chkmax (int &a , int b) { if (a<b) a=b ;}

void rf () 
{
	freopen ("bai1.inp","r",stdin) ; 
//	freopen ("bai1.out","w",stdout) ; 
}

int n , k ;
int a [maxn] ; 
int main () 
{
	ios_base::sync_with_stdio(0) ; 
	cin.tie(0) ; cout.tie(0) ; 
//	rf () ; 
	cin >> n >> k ;
	vi v ; 
	for ( int i = 1 ; i <= n ; i ++ ) 
	{
		cin >> a [i] ; 
		if ( i > 1 ) v . pb (a[i]-a[i-1]-1) ; 
	}
	
	k -- ; 
	int res = a [n] - a [1] + 1 ;  
	sort (v.rbegin(),v.rend()) ; 
	for ( int i = 0 ; i < k ; i ++ ) res -= v [i] ; 
	cout << res ; 
}

Compilation message (stderr)

stove.cpp: In function 'void rf()':
stove.cpp:33:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |  freopen ("bai1.inp","r",stdin) ;
      |  ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

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