寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #Volume21 - 2104 : Country Road

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "Country Road"
https://onlinejudge.u-aizu.ac.jp/problems/2104
カントリーロード
僕が作成、提出したコードは、以下のとおりです。

・Aizu Online Judge in C++ #Volume21 - 2104 : Country Road
/*
Aizu Online Judge in C++ #Volume21 - 2104 : Country Road
 https://onlinejudge.u-aizu.ac.jp/problems/2104
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
int main(void){
    ll t,n,k,x[100010];
    cin>>t;
    while(t--){
        cin>>n>>k;
        vector<ll> v;
        for(ll i=0;i<n;i++){
            cin>>x[i];
        }
        for(ll i=1;i<n;i++){
            v.push_back(x[i]-x[i-1]);
        }
        sort(v.rbegin(),v.rend());
        ll sum=0,p;
        for(ll i=0;i<k-1;i++){
            if(i>=v.size()){
                break;
            }
            sum+=v[i];
        }
        p=x[n-1]-x[0]-sum;
        if(p<0){
            p=0;
        }
        cout<<p<<endl;
    }
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/