【ACM】hdu1421 搬寢室 (簡單DP)

http://acm.hdu.edu.cn/showproblem.php?pid=1421


思路:

用數組a[]來保存初始數據,數組shifts[j][i]表示從一共i個數據中取出j對

首先要理解與一個物品重量差值最小的一定是與它相鄰的另一個物品,所以第一步是對這些重量排序。

接下來,我們來總結狀態轉移方程:

假設k=1,n=2.即從2個物品中拿出1對。

那麼shifts[1][2]=(a[2]-a[1])^2.

假設k=1,n=3.即從3個物品中拿出1對。

那麼shifts[1][3]=shifts[1][2]和(a[3]-a[2])^2中最小的一個。即前兩個{1,2}中最小的一對或者後兩個{2,3}中最小的一對,即最後一對。不可能是(1,3).

假設k=2,n=3.即從3個物品中拿出2對。

Impossible! shifts[2][3]=INFINITY.

假設k=1,n=4.即從4個物品中拿出1對。

那麼shifts[1][4]=shifts[1][3]和(a[4]-a[3])^2中最小的一個。即前三個{1,2,3}中最小的一對或者最後一對。

假設k=2,n=4.即從4個物品中拿出2對。

那麼shifts[2][4]=shifts[2][3]和shifts[1][2]+(a[4]-a[3])^2中最小的一個。即前三個{1,2,3}中最小的兩對或者前兩個中最小的一對加最後一對。

假設k=1,n=5.即從5個物品中拿出1對。

那麼shifts[1][5]=shifts[1][4]和(a[5]-a[4])^2中最小的一個。即前四個{1,2,3,4}中最小的一對或者最後一對。

假設k=2,n=5.即從5個物品中拿出2對。

那麼shifts[2][5]=shifts[2][4]和shifts[1][3]+(a[4]-a[3])^2中最小的一個。即前四個{1,2,3,4}中最小的兩對或者前三個中最小的一對加最後一對。

……

假設k=j,n=i.即從i個物品中拿出j對。

那麼shifts[j][i]=shifts[j][i-1]和shifts[j-1][i-2]+(a[i]-a[i-1])^2中最小的一個。即前i-1個{1,2,...,i-1}中最小的j對或者前i-2箇中最小的j-1對加最後一對。

則狀態轉移方程爲:

shifts[j][i]=min(shifts[j][i-1],shifts[j-1][i-2]+(a[i]-a[i-1])^2);

#include <iostream>
#include <string>
#include <string.h>
#include <vector>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <fstream>
using namespace std;
const int N=2005,inf=0x7fffffff;
int a[N];
int shifts[N][N];
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	a[0]=0;
	int n,k;
	while(~scanf("%d%d",&n,&k))
	{
		for (int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
		}
		sort(a+1,a+1+n);
		for (int i=0;i<=n;i++)
		{
			for (int j=1;j<=k;j++)
			{
				shifts[j][i]=inf;
			}
		}
		shifts[0][0]=0;
		for (int i=2;i<=n;i++)
		{
			for (int j=1;j*2<=i;j++)
			{
				shifts[j][i]=min(shifts[j][i-1],shifts[j-1][i-2]+(a[i]-a[i-1])*(a[i]-a[i-1]));
			}
		}
		printf("%d\n",shifts[k][n]);
	}
	//system("pause");
	return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章