HDU 1421搬寝室 swustOJ1151搬运工

HDU地址

思路，要使得差值方最小，一定是选择连续的两个数。先把数组从小到大排序。

构建出它们的差值方数组。当我们选取了一个数，那么这个数左右相邻的两个数都不能被选择。所以题意就转换为。给一个序列(我们处理好的)，求在序列中选取不相邻的k个数，使得最大值最小。

这样就可以转换为dp了。
设转移状态为dp[i][j] 表示前i个中选择j个数的最下值。
那么dp[i][j] = min(dp[i-1][j],dp[i-2][j-1] + a[i]) 表示，当前状态由当前这个不选(dp[i-1][j])转移过来,和当前状态选(dp[i-2][j-1]) 转移过来。

#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<bits/stdc++.h>
//#define int long long
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
 
const int INF = 0x3f3f3f3f;
const double eps = 1e-5;
const int mod = 998244353;
const int N = 2e3+10;

int a[N],dp[N][N];

signed main(){
	IOS
    #ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    int n,k;
    while(cin>>n>>k){
    	memset(dp,INF,sizeof(dp));
	    for(int i=1;i<=n;i++) cin>>a[i];
	    sort(a+1,a+n+1);
	    for(int i=1;i<n;i++) a[i] = (a[i+1]-a[i])*(a[i+1]-a[i]);
	    for(int i=0;i<=n;i++) dp[i][0] = 0;
	    dp[1][1] = a[1];
		for(int i=2;i<n;i++)
			for(int j=1;j<=k;j++)
				dp[i][j] = min(dp[i-1][j],dp[i-2][j-1]+a[i]);
		cout<<dp[n-1][k]<<endl;
	}
    return 0;
}