Vijos P1061 迎春舞会之三人组舞
题目
背景
HNSDFZ的同学们为了庆祝春节,准备排练一场舞
描述
n个人选出3*m人,排成m组,每组3人。
站的队形——较矮的2个人站两侧,最高的站中间。
从对称学角度来欣赏,左右两个人的身高越接近,则这一组的“残疾程度”越低。
计算公式为 h=(a-b)^2 (a、b为较矮的2人的身高)
那么问题来了。
现在候选人有n个人,要从他们当中选出3*m个人排舞蹈,要求总体的“残疾程度”最低。
格式
输入格式
第一排为m,n。
第二排n个数字,保证升序排列。
输出格式
输出最小“残疾程度”。
样例1
样例输入1
9 40
1 8 10 16 19 22 27 33 36 40 47 52 56 61 63 71 72 75 81 81 84 88 96 98 103 110 113 118 124 128 129 134 134 139 148 157 157 160 162 164
样例输出1
23
限制
各个测试点1s
提示
m<=1000,n<=5000
数据保证3*m<=n
题解
DP
f[i][j] 表示后i个人,构成j组的最小残疾程度
f[i][j]=min(f[i+1][j],f[i+2][j-1]+(a[i+1]-a[i])*(a[i+1]-a[i]))
答案储存在f[1][m]中
代码
#include<cstdio>
#include<cstring>
using namespace std;
int n,m;
int a[5005],f[5005][1005];
int readln()
{
int x=0;
char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
return x;
}
int min(int x,int y){return x<y?x:y;}
int main()
{
m=readln();n=readln();
for (int i=1;i<=n;i++) a[i]=readln();
memset(f,63,sizeof(f));
for (int i=1;i<=n;i++) f[i][0]=0;
for (int i=n-2;i>=1;i--)
{
for (int j=1;j<=min(m,(n-i+1)/3);j++)
f[i][j]=min(f[i+1][j],f[i+2][j-1]+(a[i+1]-a[i])*(a[i+1]-a[i]));
}
printf("%d",f[1][m]);
}