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;
    }
}