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/