【BZOJ4953】lydsy七月月赛 F DP

【BZOJ4953】lydsy七月月赛 F

题面

题解：设f[i][j]表示第i个强度取为j时的最小误差。那么每次转移时，我们只计算j'和j之间的像素点带来的误差，于是有：

$f[i][j]=min(f[i-1][k]+g(k..mid,k)+g(mid+1...j,j))|mid={k+jover 2}$

其中，$g(a,b)=P_a*(a-b)^2\=P_a*a*a-2*P_a*a*b+2*P_a*b*b$

于是维护P_a*a*a，P_a*a，P_a的前缀和即可。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
typedef long long ll;
ll s0[300],s1[300],s2,f[300][300],ans;
int n,m;
int main()
{
	scanf("%d%d",&n,&m);
	int i,j,k,mid;
	for(i=1;i<=n;i++)	scanf("%d",&j),scanf("%lld",&s0[j]),s1[j]=j*s0[j],s2+=j*j*s0[j];
	for(i=1;i<=255;i++)	s0[i]+=s0[i-1],s1[i]+=s1[i-1];
	memset(f,0x3f,sizeof(f));
	ans=1ll<<60;
	for(i=1;i<=m;i++)
	{
		if(i==1)	for(j=0;j<=255;j++)	f[1][j]=s0[j]*j*j-2*s1[j]*j;
		else	for(j=i-1;j<=255;j++)
		{
			for(k=i-2;k<j;k++)
			{
				mid=j+k>>1;
				f[i][j]=min(f[i][j],f[i-1][k]+(s0[mid]-s0[k])*k*k+(s0[j]-s0[mid])*j*j-2*(s1[mid]-s1[k])*k-2*(s1[j]-s1[mid])*j);
			}
		}
		if(i==m)
			for(j=i-1;j<=255;j++)	f[i][j]+=(s0[255]-s0[j])*j*j-2*(s1[255]-s1[j])*j,ans=min(ans,f[i][j]);
	}
	printf("%lld",ans+s2);
	return 0;
}