Submission #127211

#	Time	Username	Problem	Language	Result	Execution time	Memory
127211		nhimnam120	Stove (JOI18_stove)	C++14	100 / 100	88 ms	2812 KiB

stove

#include<bits/stdc++.h>
using namespace std;
vector<long long> inp;
vector<long long> pro;
signed main(){
	long long n,k;
	cin >> n >> k;
	for(int i=1;i<=n;i++){
		long long a;
		cin >> a;
		inp.push_back(a);
	}
	long long diff=1;
	sort(inp.begin(),inp.end());
	for(int i=0;i<inp.size()-1;i++){
		if(inp[i+1]-inp[i]>1){
			diff++;
			pro.push_back(inp[i+1]-inp[i]-1);
		}
	}
	long long match=diff-k;
	sort(pro.begin(),pro.end());
	if(match<=0){
		cout << n;
		return 0;
	}
	long long ans=n;
	for(int i=0;i<match;i++){
		ans+=pro[i];
	}
	cout << ans;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:15:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<inp.size()-1;i++){
              ~^~~~~~~~~~~~~

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