Submission #1837335

---

Source Code Expand

#include<bits/stdc++.h>
#define range(i,a,b) for(int i = (a); i < (b); i++)
#define rep(i,b) for(int i = 0; i < (b); i++)
#define all(a) (a).begin(), (a).end()
#define show(x)  cerr << #x << " = " << (x) << endl;
//const int INF = 1e8;
using namespace std;

#define int long long

signed main(){
	int n, k;
	cin >> n >> k;

	vector<int> v(n);
	rep(i,n){
		cin >> v[i];
	}

	sort(all(v));
	int cur = 0;
	int sum = 0;
	rep(i,k){
		sum += v[i] + cur;
		cur++;
	}
	cout << sum << endl;
	return 0;

	cout << accumulate(v.begin(), next(v.begin(), k), 0) + (k - 1) * k / 2 << endl;
}

Submission Info

Judge Result