n(n≤4000)
个人排队要乘
k(k≤min(n,800))
艘船,每艘船坐的人数不限。第i个人对第j个人有一个不满意度
Ui,j(0≤Ui,j≤9且Ui,j=Uj,i=0)
,如果他们在一条船中的话。问,如何安排使得总不满度最小。
DP,满足四边形不等式。
f(i,j)
表示前面j个人乘坐i只船的不满度,那么
f(i,j)=min(f(i−1,k)+w(k+1,j)|i−1≤k<j)
,其中
w(k+1,j)
是第k+1个人到第j个人在同一艘船中的不满度,通过
O(n2)
的时间复杂度求出来。
#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
void GET(int &t)
{
char c; t = 0;
do{c=getchar();}while(!isdigit(c));
while(isdigit(c)){t=t*10+c-'0';c=getchar();}
}
int a[4010][4010],w[4010][4010],n,m,f[4010][810],s[4010][810];
int main()
{
GET(n);GET(m);
for(int i = 1; i <= n; i++)
for(int j = 1; j <= n; j++)
GET(a[i][j]),f[i][j] = 0x6f7f7f7f;
for(int i = 1; i <= n; i++)
for(int j = 1; j <= n; j++)
a[i][j] = a[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1];
for(int i = 1; i <= n; i++)
for(int j = i+1; j <= n; j++)
w[i][j] = (a[j][j]-a[i-1][j]-a[j][i-1]+a[i-1][i-1])/2;
for(int i = 1; i <= n; i++) f[1][i] = w[1][i],s[1][i] = 0;
for(int i = 2; i <= m; i++)
{
s[i][n+1] = n;
for(int j = n; j >= i; j--)
for(int k = s[i-1][j]; k <= s[i][j+1]; k++)
if(f[i-1][k]+w[k+1][j] < f[i][j])
f[i][j] = f[i-1][k]+w[k+1][j],s[i][j] = k;
}
printf("%d\n",f[m][n]);
}