fork download
copy 
  1. #include <bits/stdc++.h>
  2. #include<iostream>
  3. #include<algorithm>
  4. using namespace std;
  5.  
  6.  
  7.  
  8.  
  9. int main() {
  10.         ios_base::sync_with_stdio(false);
  11.            cin.tie(NULL);
  12.  
  13.     int t;cin>>t;
  14.     while(t--){
  15.         int n,k;cin>>n>>k;
  16.         long long sum = 0;
  17.         vector<int>v(n+1);
  18.         for(int i =1;i<=n;i++){
  19.             cin>>v[i];
  20.             sum+=v[i];
  21.         }
  22.         vector<long long>dif(n+1);
  23.         long long mx = 0 , mxINd=0;
  24.         for(int i = 2;i<=n;i++){
  25.             dif[i] = abs(v[i]-v[i-1]);
  26.             if(dif[i]>mx){
  27.                 mx = max(mx , dif[i]);
  28.                 mxINd = i;
  29.             }
  30.         }
  31.  
  32.         while(k--){
  33.             sum-=mx;
  34.             // cout<<" mx "<<mx<<endl;
  35.             v[mxINd] = min(v[mxINd] , v[mxINd-1]);
  36.             v[mxINd-1] = min(v[mxINd] , v[mxINd-1]);
  37.             mx = 0;
  38.             for(int i = 2;i<=n;i++){
  39.             dif[i] = abs(v[i]-v[i-1]);
  40.             if(dif[i]>mx){
  41.                 mx = max(mx , dif[i]);
  42.                 mxINd = i;
  43.             }
  44.             }
  45.             // for(auto a : v)cout<<a<<" ";cout<<endl;
  46.         }
  47.         cout<<sum<<endl;
  48.  
  49.     }
  50.  
  51.  
  52.  
  53.  
  54.  
  55. }
Success #stdin #stdout 0s 5304KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
Standard output is empty
https://ideone.com/yJ3105
language:
C++ (gcc 8.3)
created:
16 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!