Submission #152837

#	Time	Username	Problem	Language	Result	Execution time	Memory
152837		erebos	Wine Tasting (FXCUP2_wine)	C++17	1 / 1	78 ms	5356 KiB

wine

#include <bits/stdc++.h>
using namespace std;
using ll=long long;

vector<int>arr;

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    int n, k;
    cin>>n>>k;
    for(int i=0; i<n; ++i) {
        int a;
        cin>>a;
        arr.push_back(a);
    }
    sort(arr.begin(), arr.end());
    ll ans=0;
    if(k%2==0) {
        for(int i=0; i<k/2; ++i) {
            ans+=arr[n-i-1];
        }
        for(int i=0; i<k/2-1; ++i) {
            ans-=arr[i];
        }
    } else {
        for(int i=0; i<k/2+1; ++i) {
            ans+=arr[n-i-1];
        }
        for(int i=0; i<k/2; ++i) {
            ans-=arr[i];
        }
    }
    cout<<ans;
}

• Subtask 1: 1 / 1