#include "bits/stdc++.h"
using namespace std;
#define int long long
signed main () {
int n, k;
cin >> n >> k;
vector <int> a(n);
for(int i = 0; i < n; i ++) cin >> a[i];
function <int(int,int)> good=[&](int x, int opt) {
int cnt = 1, ff = a[0], ans = 0;
for(int i = 1; i < n; i ++) {
if(a[i]-ff > x) {
cnt ++;
ans += a[i-1]-ff+1;
ff = a[i];
}
}
ans += a[n-1]-ff+1;
if(opt) return ans;
if(cnt <= k) return 1;
return 0;
};
int l = 0, r = 1e9;
while(l < r) {
int mid = (l + r) / 2;
if(good(mid, 0)) r = mid;
else l = mid + 1;
}
cout << good(r, 1);
return 0;
}
Compilation message
stove.cpp: In lambda function:
stove.cpp:21:23: error: inconsistent types 'long long int' and 'int' deduced for lambda return type
21 | if(cnt <= k) return 1;
| ^
stove.cpp:22:10: error: inconsistent types 'long long int' and 'int' deduced for lambda return type
22 | return 0;
| ^