动态规划题
题意是输入两个整数n和k,表示从n件物品中搬走k对(每次左右手各搬一件),已知每件物品的疲劳值,求搬走k对后的最小疲劳度。
每个物品的疲劳度用ans存起来,很明显,把ans按升序排序后每相邻两个物品的疲劳度都是最小的;
然后定义数组dp【i】【j】表示从前i件物品中搬走j对的总疲劳度,则对于第i件物品:
(1)如果搬走第i件,那么一定也要搬走第i-1件物品,不然第i件物品和谁配对呢,同时从前i-2件物品中搬走j-1对,
此时dp【i】【j】=dp【i-2】【j-1】+(ans【i】-ans【i-1】)^2;
(2)如果不搬走第i件物品,那么就等于从前i-1件物品里面搬走j对物品了,即dp【i】【j】=dp【i-1】【j】;
显然状态转移方程:dp【i】【j】=min(dp【i-2】【j-1】+(ans【i】-ans【i-1】)^2,dp【i-1】【j】);
#include <cstdio>
#include <algorithm>#include <iostream>
#define MAX 2100
#define INF 999999999
using namespace std;
int ans[MAX],dp[MAX][MAX/2];
int main()
{
int n,k;
while(cin>>n>>k)
{
for(int i=1;i<=n;i++)
scanf("%d",&ans[i]);
sort(ans+1,ans+1+n);
for(int i=1;i<=n;i++)
for(int j=0;j<=k;j++)
dp[i][j]=INF;
for(int i=0;i<=n;i++)
dp[i][0]=0;
for(int i=2;i<=n;i++)
for(int j=1;j<=k&&2*j<=i;j++)
dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(ans[i]-ans[i-1])*(ans[i]-ans[i-1]));
cout<<dp[n][k]<<endl;
}
return 0;
}