#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef complex<ld> point;
void debug_out() {cerr << endl;}
template<typename Head, typename... Tail>
void debug_out(Head H, Tail... T){
cerr << H << ' ';
debug_out(T...);
}
#define debug(...) cerr << "(" << #__VA_ARGS__ << "): ", debug_out(__VA_ARGS__)
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define MP(x, y) make_pair(x, y)
const int maxn = 1e5 + 10;
int val[maxn];
long long take_photos(int n, int m, int k, std::vector<int> r, std::vector<int> c) {
sort(all(r));
for (int i = 1; i < n; i++){
val[i] = r[i] - r[i-1];
}
sort(val + 1, val + n + 1);
ll ans = 0;
for (int i = 1; i <= n-k; i++){
ans += 1ll * (val[i]+1) * (val[i]+1) - 2;
}
return ans;
}
