Submission #754616

#	Time	Username	Problem	Language	Result	Execution time	Memory
754616		Bula	Stove (JOI18_stove)	C++14	100 / 100	52 ms	2232 KiB

stove

#include <bits/stdc++.h>
using namespace std;
 
#define ll long long
#define pb push_back
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define int ll
 
const ll mod=1e9+7;
 
main(){
 
    int tt=1;//cin>>tt;
    while(tt--){
    	int n,k;
    	cin>>n>>k;
    	vector<int> v(n),diff;
    	
    	for(int i=0;i<n;i++){
    		cin>>v[i];
		}
		
		for(int i=1;i<n;i++){
			diff.pb(v[i]-v[i-1]);
		}
		
		sort(rall(diff));
		int N=diff.size(),t=0;
		for(int i=0;i<min(N,k-1);i++){
			t+=diff[i];
		}
		if(k==n) cout<<n<<endl;
		else cout<<(v[n-1]-v[0])-t+k<<endl;
		//for(int i=0;i<n;i++) cout<<p[i]<<" ";
	}
 
}
 
 

Compilation message (stderr)

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

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