fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. int n, k;
  6. cin >> n >> k;
  7. vector<int> a(n);
  8. for (int i = 0; i < n; i++){
  9. cin >> a[i];
  10. }
  11.  
  12. vector<int> gaps;
  13. for (int i = 1; i < n; i++){
  14. gaps.push_back(a[i] - a[i-1] - 1);
  15. }
  16.  
  17. sort(gaps.begin(), gaps.end());
  18.  
  19. int ans = n;
  20. for (int i = 0; i < n - k; i++){
  21. ans += gaps[i];
  22. }
  23. cout << ans << '\n';
  24. }
Success #stdin #stdout 0s 5316KB
stdin
3 2
1
3
6
stdout
4