제출 #1162610

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1162610		Vidicy	Stove (JOI18_stove)	C++17	100 / 100	13 ms	1864 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pll pair<long long, long long>
#define st string
#define fr first
#define sec second
#define pb push_back
#define beg begin
#define INF 1e18
#define MOD 1e9+7


void solve() {

    int n,k;
    cin >> n >> k;
    int cnt = n-(n-k);
    vector<int> a(n);
    for(int i = 0; i < n; i ++)cin >> a[i];
    if(n == k) cout << k;
    else{
        vector<int> b(n-1);
        for(int i = 0; i < n-1; i++){
            b[i] = abs(a[i] - a[i+1]);
        }

        sort(b.begin(),b.end());
        for(int i = 0; i < n-k; i++){
            cnt = cnt + b[i];
        }
        cout << cnt << endl;
    }

}



signed main() {
    ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);cout << setprecision(20);
    //    freopen("input.txt", "r", stdin);
    //    freopen("output.txt", "w", stdout);
    int T = 1;
    //cin >> T;
    while(T--) {
        solve();
    }

}

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