Submission #868904

#	Time	Username	Problem	Language	Result	Execution time	Memory
868904		autoastt	Stove (JOI18_stove)	C++14	0 / 100	1 ms	348 KiB

stove

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int a[N], b[N], ans, cnt;
 
int main(){
 int n, k;
  cin >> n >> k;
  vector<pair<int,pair<int,int>>> v;
  for(int i=0;i<n;i++){
cin >> a[i];
    if(i>0) v.push_back({a[i]-a[i-1]+1,{i,i-1}});
}
  sort(v.begin(),v.end());
  for(int i=0,j=0;i<v.size();i++){
ans+=v[i].first;
    j++;
	//cout << j << ' ' << ans << "\n";
    if(!b[v[i].second.first]) b[v[i].second.first]=1,cnt++;
      if(!b[v[i].second.second]) b[v[i].second.second] = 1,cnt++;
    if(n-cnt+j==k){
cout << ans+j;
return 0;
}
}
 
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:15:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |   for(int i=0,j=0;i<v.size();i++){
      |                   ~^~~~~~~~~

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