Submission #708439

#	Time	Username	Problem	Language	Result	Execution time	Memory
708439		deme_bz	Stove (JOI18_stove)	C++14	0 / 100	1 ms	340 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){
		for(int i=v.size()-1;i>=v.size()-1-(k-1)+1;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"
      |                                ^~~~~~~~~~~~
stove.cpp: In function 'void test_case()':
stove.cpp:31:25: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |   for(int i=v.size()-1;i>=v.size()-1-(k-1)+1;i--){
      |                        ~^~~~~~~~~~~~~~~~~~~~
stove.cpp: At global scope:
stove.cpp:42:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   42 | main(){
      | ^~~~

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