Submission #708454

#	Time	Username	Problem	Language	Result	Execution time	Memory
708454		deme_bz	Stove (JOI18_stove)	C++14	100 / 100	87 ms	2240 KiB

stove

#pragma GCC diagnostic warning "-std=c++11"
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define ff first
#define ss second
#define eb emplace_back
#define sz(x) (int)x.size()
using namespace std;
const int N=2e5+5;
 
ll A[N];
void test_case(){
	//cout << INT_MAX+1 << endl; 
	int n,k;
	cin >> n >> k;
	
	for(int i=1;i<=n;i++){
		cin >> A[i];
	}
	vector<ll> v;
	for(int i=2;i<=n;i++){
		v.pb(A[i]-A[i-1]-1);
	}
		sort(all(v));
	
	ll answer=A[n]-A[1]+1;
	//cout << v.size()-1 << endl;
	if(v.size()>0){
		assert(k-1<=v.size());
		for(int i=v.size()-1;i>=0 && i>=v.size()+1-k;i--){
			answer-=v[i];
		}
	}
	
	cout << answer << endl;
}




main(){
	//ios_base::sync_with_stdio(false);
    int T = 1;
    //cin >> T;
    while(T--){
        test_case();
    }
}

Compilation message (stderr)

stove.cpp:1:32: warning: '-std=c++11' is not an option that controls warnings [-Wpragmas]
    1 | #pragma GCC diagnostic warning "-std=c++11"
      |                                ^~~~~~~~~~~~
In file included from /usr/include/c++/10/cassert:44,
                 from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:33,
                 from stove.cpp:2:
stove.cpp: In function 'void test_case()':
stove.cpp:31:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |   assert(k-1<=v.size());
      |          ~~~^~~~~~~~~~
stove.cpp:32:33: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |   for(int i=v.size()-1;i>=0 && i>=v.size()+1-k;i--){
      |                                ~^~~~~~~~~~~~~~
stove.cpp: At global scope:
stove.cpp:43:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   43 | main(){
      | ^~~~

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