关于搬寝室问题自己的看法

自学算法有一个月，动态规划是真的难，本来有点想放弃，不过最近边打游戏边学习，动态规划有了一点眉目。本人一个小菜鸟，研究这个搬寝室问题研究了一下午，总算弄懂了。

/*
* 有n个行李，每个行李有一个重量。
* 现在你要搬走2k个行李，你一共去k次。每次左手右手各
* 拿一个行李，假设这两个行李的重量分别是x和y。
* 那么这一次搬运产生的疲惫度是(x-y)^2
* 现在你希望最小化疲惫度
* 2<=2k<=n<2000
*/
#include<iomanip>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int>luggage = { 7,4,1,9,15,17 };//行李
inline int Fatigue(int a, int b) { return (luggage[a] - luggage[b]) * (luggage[a] - luggage[b]); };
int main()
{
	int n = (int)luggage.size();
	/*先把行李按照从小到大的顺序排序，这样一次
	  取相邻的两个就能使得相邻的两个行李重量之
	  差最小*/
	sort(luggage.begin(), luggage.end(), [](int a, int b) {return a < b; });
	int k;
	cout << "请输入要搬得次数: ";
	cin >> k;
	while (cin.fail() || k <= 0 || k > n / 2)
	{
		cin.clear();
		cin.ignore(512, '\n');
		cout << "请输入要搬得次数: ";
		cin >> k;
	}
	
	/*choice和f(choice)[i][j]表示从第i个行李到第j个行李选出choice对的最小疲惫度。
	  choice=0的时候，f(0)[i][j]=0。
	  choice=1的时候，j<=i时，f(1)[i][j]=0;
					  j=i+1时，f(1)[i][j]=Fatigue(i,j);
					  其他情况时，如果选择了第j个行李，那么第j-1个行李应该和第j个行李一起拿走,则f(1)[i][j]=f(0)[i][j-2]+Fatigue(j-1,j);
					              如果没有选择第j个行李，那么f(1)[i][j]=f(1)[i][j-1];
								  因此f(1)[i][j]=min(f(0)[i][j-2]+Fatigue(j-1,j)，f(1)[i][j-1])。
	  choice=2的时候，j<=i+2时，f(2)[i][j]=0;
					  j=i+3时，f(2)[i][j]=Fatigue(i,i+1)+Fatigue(i+2,j);
					  其他情况时，如果选择了第j个行李，那么第j-1个行李应该和第j个行李一起拿走，则f(2)[i][j]=f(1)[i][j-2]+Fatigue(j-1,j);
								  如果没有选择第j个行李，那么f(2)[i][j]=f(2)[i][j-1];
								  因此f(2)[i][j]=min(f(1)[i][j-2]+Fatigue(j-1,j)，f(2)[i][j-1])。
	  choice>0的时候，j<=i+(choice-1)*2时，f(choice)[i][j]=0;
					 j=i+choice*2-1时，f(choice)[i][j]=Fatigue(i,i+1)+Fatigue(i+2,i+3)+...+Fatigue(j-1,j);
					 其他情况时，如果选择了第j个行李，那么第j-1个行李应该和第j个行李一起拿走，则f(choice)[i][j]=f(choice-1)[i][j-2]+Fatigue(j-1,j);
								 如果没有选择第j个行李，那么f(choice)[i][j]=f(choice)[i][j-1];
								 因此f(choice)[i][j]=min(f(choice-1)[i][j-2]+Fatigue(j-1,j)，f(choice)[i][j-1])。
	  最终结果是f(choice)[0][n-1]
	  这样做需要两张表格交替更新，并且还要对choice进行一次循环，时间复杂度O(n^3)。
	  改进:f[i][j]表示前i个行李选出了j对行李的最小疲惫度。(0<=i<n+1,0<=j<k+1)
	  这样设计，填充表格不再是对角线填充，而是纵向填充，最终结果是f[n][k];
	  1、当i=0或者i=1时，即行李小于2个，此时无法选择，f[i][j]=0;
	  2、当j=0时，不选择行李,f[i][j]=0;
	  3、当i=2, j=1时，即选择前两个行李，f[i][j]=Fatigue(i-1,i);
	  4、当2<i<=n，j=1时，即前i个行李选择1对出来，
						  如果选择第i个行李，那么一定要跟第i-1个行李一起拿走，则f[i][1]=f[i-2][0]+Fatigue(i-1,i);
						  如果不选择第i个行李，则f[i][1]=f[i-1][1];
						  所以f[i][1]=min(f[i-2][0]+Fatigue(i-1,i),f[i-1][j])。
	  5、其他情况，即前i个行李选择j对出来，
						  如果选择第i个行李，那么一定要跟第i-1个行李一起拿走，则f[i][j]=f[i-2][j-1]+Fatigue(i-1,i);
						  如果不选择第i个行李，则f[i][j]=f[i-1][j];
						  所以f[i][j]=min(f[i-2][j-1]+Fatigue(i-1,i),f[i-1][j])。
	  特别的:理论上i>=2*j,如果小于，则全选，即f[i][j]=f[i][j-1];
			 但是i=2*j时，也必须全部选择，即f[i][j]=f[i-2][j-1]+Fatigue(i-1,i)，这个可以与3合并;
		据观察，4可以合并到5中
	*/
	int** f = new int*[n+1];
	for (int i = 0; i <= n; i++)
		f[i] = new int[k+1] {0};

	for (int j = 1; j <= k; j++)
	{
		for (int i = 2; i <= n; i++)
		{
			if (i ==2 * j)
				f[i][j] =f[i-2][j-1] + Fatigue(i - 2, i - 1);
			else if (i < 2 * j)
				f[i][j] = f[i][j - 1];
			else
				f[i][j] = min(f[i - 2][j - 1] + Fatigue(i - 2, i-1), f[i - 1][j]);//这里注意Fatigue的下标
		}
	}

	//测试
	//for (int i = 0; i <= n; i++)
	//{
	//	for (int j = 0; j <= k; j++)
	//		cout << setw(3) << f[i][j];
	//	cout << endl;
	//}

	cout << f[n][k] << endl;

	for (int i = 0; i <= n; i++)
		delete[] f[i];
	delete[] f;
	return 0;
}

如果要想追踪选择了哪些行李，不知道各位大神有没有好的解法？