题目传送门:hdu1421
解题思路:
一道DP题
给定n个物品,每个物品有重量,
从中选出m对,使得这m对物品重量差的平方和最小。
疲劳度:m对物品重量差的平方和
分析与解题思路
先对n中物品的重量排序
令dp[i][j]表示前i个物品中选j对的最小疲劳度。
则dp[i][j]可能含有第i个物品(这种情况下,第i种物品一定是和第i-1个物品配对),
则dp[i][j]=dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1])
dp[i][j]的j对也可能不含有第i个物品,此时有
dp[i][j]=dp[i-1][j]
状态转移方程
dp[i][j]=min{dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1]),dp[i-1][j]
#include<stdio.h> #include<stdlib.h> #include<malloc.h> #include<string.h> #include<string.h> #include<algorithm> using namespace std; int n,k; long long dp[2005][1005]; long long p[2005]; long long min (long long a ,long long b ) { return a<b?a:b; } int cmp(const void* a,const void* b) { return *(int*)a-*(int*)b; } int main() { while (scanf ("%d%d",&n,&k) != EOF) { for (int i =1 ;i<=n;i++) scanf ("%I64d",&p[i]); qsort (p+1,n,sizeof (p[0]),cmp); for (int i =1 ;i<=n;i++) for (int j=1 ;j<=k;j++) { dp[i][j]=2147483646 ; } for (int i = 2 ;i<=n;i++ ) { for (int j=1 ;j*2<=i;j++) { dp[i][j] = min ( dp[i-1][j],dp[i-2][j-1] + (p[i]-p[i-1])*(p[i]-p[i-1]) ); } } printf("%I64d\n",dp[n][k]); } return 0; }