Submission #365991

#	Time	Username	Problem	Language	Result	Execution time	Memory
365991		vishesh312	Stove (JOI18_stove)	C++17	50 / 100	1091 ms	5476 KiB

stove

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl "\n"
void solve(){
    int number = 0;
    cerr << number << '\n'; ++number;
  int tc,k;
  cin>>tc>>k;
  k--;
  vector<int>v,diff;
  for(int i=0;i<tc;i++){
    int n;
    cin>>n;
    v.push_back(n);
    v.push_back(n+1);
  }
    cerr << number << '\n'; ++number;
  int ans=0;
  ans+=tc;
    cerr << number << '\n'; ++number;
  auto st = v.begin();
  auto end = v.end();
  end--;
  v.erase(end);
  v.erase(st);
    cerr << number << '\n'; ++number;
  for(int i=0;i<v.size();i+=2){
    diff.push_back(v[i+1]-v[i]);
  }
    cerr << number << '\n'; ++number;
  sort(diff.rbegin(),diff.rend());
    cerr << number << '\n'; ++number;
  while(k!=0){
    cerr << number << '\n'; ++number;
    auto it = diff.begin();
    diff.erase(it);
    k--;
  }
    cerr << number << '\n'; ++number;
  for(int i=0;i<diff.size();i++){
    ans+=diff[i];
    cerr << number << '\n'; ++number;
  }
    cerr << number << '\n'; ++number;
  cout<<ans;
}
signed main() {
  ios_base::sync_with_stdio(false); cin.tie(NULL);
  int tc=1;
  for(int i=0;i<tc;i++)solve();
  return 0;
}

Compilation message (stderr)

stove.cpp: In function 'void solve()':
stove.cpp:28:16: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |   for(int i=0;i<v.size();i+=2){
      |               ~^~~~~~~~~
stove.cpp:41:16: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |   for(int i=0;i<diff.size();i++){
      |               ~^~~~~~~~~~~~

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