#include "aliens.h"
#include <cstdio>
#include <cassert>
#include <vector>
#include <iostream>
using namespace std;
int maxim[100005];
int until[100005];
int maximpref[100005];
long long dp[10005][1005];
long long take_photos(int n, int m, int k, std::vector<int> r, std::vector<int> c) {
swap(n,m);
for(int i=0;i<=n;i++)
{
maxim[i]=n+1;
until[i]=n+1;
maximpref[i]=0;
}
for(int i=0;i<m;i++)
{
r[i]++;
c[i]++;
maxim[max(r[i],c[i])]=min(maxim[max(r[i],c[i])],min(r[i],c[i]));
maximpref[min(r[i],c[i])]=max(maximpref[min(r[i],c[i])],max(r[i],c[i]));
}
for(int i=1;i<=n;i++)
{
if(maximpref[i]>maximpref[i-1])
{
for(int j=maximpref[i-1]+1;j<=maximpref[i];j++)
{
until[j]=i;
}
}
maximpref[i]=max(maximpref[i],maximpref[i-1]);
}
//cout<<'\n';
for(int i=1;i<=n;i++)
{
for(int j=0;j<=k;j++)
{
dp[i][j]=1ll*n*n;
if(maxim[i]==n+1)
{
dp[i][j]=dp[i-1][j];
continue;
}
if(j==0)
{
continue;
}
for(int lower=min(until[i],maxim[i]);lower>=1;lower--)
{
int nxt=maximpref[lower-1];
long long add;
if(nxt>=lower)
{
add=1ll*(nxt-lower+1)*(nxt-lower+1);
}
else
{
add=0;
}
dp[i][j]=min(dp[i][j],dp[nxt][j-1]+1ll*(i+1)*(i+1)+1ll*lower*lower-2*(i+1)*lower-add);
}
}
}
return dp[n][k];
}
