【问题描述】
背景 Background
HNSDFZ的同学们为了庆祝春节,准备排练一场舞
【描述 Description】
n个人选出3*m人,排成m组,每组3人。
站的队形——较矮的2个人站两侧,最高的站中间。
从对称学角度来欣赏,左右两个人的身高越接近,则这一组的“残疾程度”越低。
计算公式为 h=(a-b)^2 (a、b为较矮的2人的身高)
那么问题来了。
现在候选人有n个人,要从他们当中选出3*m个人排舞蹈,要求总体的“残疾程度”最低。
【输入格式 Input Format】
第一排为m,n。
第二排n个数字,保证升序排列。
【输出格式 Output Format】
输出最小“残疾程度”。
【样例输入 Sample Input】
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
【样例输出 Sample Output】
23
注释 Hint
m<=1000,n<=5000
数据保证3*m<=n
【算法分析】
从大到小排序 a[i]
f[i,j] 表示前i个数中有j对跳舞组合时的最优解
应为要最优就必须是排序时相邻两数的在两边才最好
而中间的人最高,记 f[i,j]=f[i-2,j-1]+(a[i]-a[i-1])^2
如果不取a[j]就 记 f[i,j]=f[i-1,j]
所以 f[i,j]=min{f[i,j]=f[i-2,j-1]+(a[i]-a[i-1])^2 | f[i,j]=f[i-1,j]}
huyichen
摘自一大牛语录:
首先将筷子长度从短到长排序。
F[i,j,0]表示i个人使用前j双筷子,且第j根筷子不用,所需要长度差的平方和的最小值。
F[i,j,1]表示i个人使用前j双筷子,且使用第j根筷子,所需要长度差的平方和的最小值。
则F[i,j,0]=min{f[i,j-1,1],f[i,j-1,0]},F[i,j,1]=F[i-1,j-1,0]+(l[j]-l[j-1])^2
ans=min{F[k,n,1],F[k,n,0]}
算法复杂度为O(NK)。
算法正确性的简单证明:
因为筷子配对的时候要求是长度平方差最小,所以每根筷子的配对的时候总是希望和长度差最小的配对,即排序后相邻两根筷子是可以配对的。l[a],l,l[c],l[d]表示4根排序好的筷子。因为(l-l[a])^2+(l[d]-l[c])^2<(l-l[c])^2+(l[d]-l[a])^2,所以上述配对方法是正确的。
本题同样可以把模型直接转变为二分图最小权匹配。
将两根筷子相连的边权值标记为两根筷子长度平方差,然后对二分图求一次最小权匹配。可以使用网络流的最小费用K流的算法,寻找K次最短路的增广轨。
算法复杂度为O(KN^3)。
非常显然,动态规划的复杂度远远低于二分图最小权匹配的复杂度。
【代码】
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | #include <stdio.h> #include <string.h> int a[1001][5001]; int b[5001]; int main() { int n, m, i, j, k, temp; scanf("%d%d", &m, &n); memset(b, 0, sizeof(b)); for (i = 1; i <= n; i++) scanf("%d", &b[i]); for (i = 0; i <= m; i++) { for (j = 0; j <= n; j++) a[i][j] = 2000000000; } for (i = 0; i <= n; i++) a[0][i] = 0; for (i = 1; i <= m; i++) { for (j = i + i; j <= n - (m - i) * 3; j++) { a[i][j] = a[i][j - 1]; temp = a[i - 1][j - 2] + (b[j] - b[j - 1]) * (b[j] - b[j - 1]); if ( j < n - (m - i) * 3 && temp < a[i][j]) a[i][j] = temp; } } printf("%d\n", a[m][n]); return 0; } |