牛客网 wannfly练习赛7 B 购物 dp

思路:dp[i][j]表示第i天有j个糖果的最小花费。过%60数据的可能是RE,数组开大一点会MLE,所以需要优化对于每一天买的糖果不需要超过n个糖果。记录前缀和,然后用类似背包方法从第i-1天递推第i天。

#include<iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 305;
int dp[MAXN][MAXN];
int a[MAXN][MAXN];
int main()
{
    int n, m;
    scanf("%d %d", &n, &m);
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j)
            scanf("%d",&a[i][j]);
    for(int i=1;i<=n;++i)
        sort(*(a+i)+1,*(a+i)+m+1);
    for(int i=1;i<=n;++i)
    {
        for(int j=1;j<=m;++j)
            a[i][j]=a[i][j-1]+a[i][j];
    }
    int k=2;
    memset(dp,0x3f,sizeof(dp));
    for(int i=1;i<=m;++i)
        dp[1][i]=a[1][i]+i*i;
    for(int i=2;i<=n;++i)
    {
        for(int j=i;j<=k*m;++j)
        {
            dp[i][j]=min(dp[i-1][j],dp[i][j]);

            for(int c=1;c<=m;++c)
            {
                if(j-c>=i-1)
                dp[i][j]=min(dp[i][j],dp[i-1][j-c]+c*c+a[i][c]);
            }
        }
        k++;
    }
    printf("%d\n", dp[n][n]);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值