购物----------------------------思维(dp)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

解析:
因为题目要求价格最低,所以我们需要给m个糖果从小到大排序,然后再计算出每一天的m个糖果的价格前缀和
设f[i][j]:表示第i天,买了j个糖果

状态转移方程:f[i][j]=min(f[i][j],f[i-1][k]+sum[i][j-k]+(j-k)*(j-k));

第一维(第一重循环): 枚举第i天
第二维(第二重循环): 枚举第i天选的糖果个数(这里没判断每天选的糖果个数不能超过m个,因为上面加了一个初始化)。第i天选的糖果个数起始一定是i,因为前i天至少选了i个。

第三重循环: 枚举i-1天选了糖果的个数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF=1e9;
const int N=550;
int a[N][N];
ll sum[N][N];
ll f[N][N],n,m;
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=0;i<=n;i++)//此处初始化会抵消选糖果数超过m个的情况
	{
		for(int j=1;j<=400;j++) f[i][j]=sum[i][j]=INF;
	}
	for(int i=1;i<=n;i++)//处理前缀和
	{
		for(int j=1;j<=m;j++) cin>>a[i][j];
		sort(a[i]+1,a[i]+1+m);
		for(int j=1;j<=m;j++) sum[i][j]=sum[i][j-1]+a[i][j];
		
	}
	//初始化赋值第一天选m个糖果的情况
	for(int i=1;i<=m;i++) f[1][i]=sum[1][i]+i*i;
	for(int i=2;i<=n;i++)
	{
		for(int j=i;j<=n;j++)//j=i,因为枚举第i天选的,因为钱i-1天已经至少选了i-1个,所以j==i
		{
			for(int k=i-1;k<=j;k++)//k=i-1 因为枚举第i天正要选的,所以已经选好了i-1个
				f[i][j]=min(f[i][j],f[i-1][k]+sum[i][j-k]+(j-k)*(j-k));
		}
	}
	cout<<f[n][n]<<endl;
}

解析:
不用初始化限制只能选m个
在循环里加限制

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF=1e9;
const int N=550;
int a[N][N];
ll sum[N][N];
ll f[N][N],n,m;
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++) cin>>a[i][j];
		sort(a[i]+1,a[i]+1+m);
		for(int j=1;j<=m;j++) sum[i][j]=sum[i][j-1]+a[i][j];
		
	}
	memset(f,0x3f,sizeof f);
//	for(int i=1;i<=m;i++) f[1][i]=sum[1][i]+i*i;
	f[0][0]=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j<=n;j++)
		{
			for(int k=0;k<=j&&k<=m;k++)
			{
			//	if(f[i-1][j-k]!=0x3f3f3f3f)
				//{
					f[i][j]=min(f[i][j],f[i-1][j-k]+sum[i][k]+k*k);
			//	}
				
			}
			
		}
	}
	cout<<f[n][n]<<endl;
}
发布了491 篇原创文章 · 获赞 11 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_43690454/article/details/104839738