购物(dp)

题目链接

认真读题，认真读题，认真读题！！！

道路千万条，读题第一条，读题不仔细，wa两行泪！

这是一个01背包的变形，三个循环，第一个枚举天数，第二个枚举前 i 天买到的糖果，第三个枚举今天买走的糖的数量。

坑点：
题目描述中明确说明要求n天中每天都能吃到糖果，所以前 i 天至少买到了 i 个糖果，所以第二层循环下标应该从 i 开始。

初始化问题:
第0天买0个花销为0，第0天无法购买多余0个，且第 i 天之前买的不能少于 i 个，所以dp数组的第一行第一列不用初始化，保持一个较大的值即可，只需要将dp[0][0]赋值为0即可。

#include<bits/stdc++.h>
using namespace std;
int arr[330][330];
int dp[330][330];
int sum[330][330];
int cost, tcost, n, m;
int main()
{
	scanf("%d %d", &n, &m);
	memset(dp, 0x3f, sizeof dp);
	int price;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++)scanf("%d", &arr[i][j]);
		sort(arr[i] + 1, arr[i] + m + 1);
	}
	for (int i = 1; i <= n; i++) {
		sum[i][0] = 0;
		for (int j = 1; j <= m; j++) {
			sum[i][j] = sum[i][j - 1] + arr[i][j];
		}
	}
	dp[0][0] = 0;
	for (int i = 1; i <= n; i++) {
		for (int j = i; j <= min(n, i*m); j++) {
			for (int k = 0; k <= m; k++) {
				if (j >= k)dp[i][j] = min(dp[i][j], dp[i - 1][j - k] + k * k + sum[i][k]);
			}
		}
	}
	printf("%d", dp[n][n]);
	return 0;
}