HDOJ 1421 搬寝室 杭电 ACM

此类状态转移方程在DP问题中很常见

这题先要理解 为何每趟只能搬排序后相邻的物品 PPT中给出的不等式可自证 有助于理解

#include<algorithm>#include<iostream>using namespace std;int n,k,i,j,t,a[2001],f[2001][1001];//f[x][y]保存x件物品中搬y趟(共2y件)的最优解int main(){    while(cin>>n>>k){        for(i=1;i<=n;++i)cin>>a[i];        sort(a+1,a+n+1);//根据之前的分析 需要排序        for(f[0][0]=0,i=1,t=2;i<=k;++i,t=i*2){//i代表趟数            f[t][i]=f[t-2][i-1]+(a[t]-a[t-1])*(a[t]-a[t-1]);//i趟正好搬完的情况            for(j=t+1;j<=n;++j)//i趟下 推出增加物品总数后的最优解 用来做趟数增加后运算的依据                  f[j][i]=min(f[j-1][i],f[j-2][i-1]+(a[j]-a[j-1])*(a[j]-a[j-1]));//每趟都要相邻搬 只有俩情况 取较小的        }        cout<<f[n][k]<<endl;    }}