牛客—购物

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

在遥远的东方，有一家糖果专卖店。

这家糖果店将会在每天出售一些糖果，它每天都会生产出m个糖果，第i天的第j个糖果价格为C[i][j]元。

现在的你想要在接下来的n天去糖果店进行选购，你每天可以买多个糖果，也可以选择不买糖果，但是最多买m个。（因为最多只生产m个）买来糖果以后，你可以选择吃掉糖果或者留着之后再吃。糖果不会过期，你需要保证这n天中每天你都能吃到至少一个糖果。

这家店的老板看你经常去光顾这家店，感到非常生气。（因为他不能好好睡觉了）于是他会额外的要求你支付点钱。具体来说，你在某一天购买了 k 个糖果，那么你在这一天需要额外支付 k2 的费用。

那么问题来了，你最少需要多少钱才能达成自己的目的呢？

输入描述:

第一行两个正整数n和m，分别表示天数以及糖果店每天生产的糖果数量。
接下来n行（第2行到第n+1行），每行m个正整数，第x+1行的第y个正整数表示第x天的第y个糖果的费用。

输出描述:

输出只有一个正整数，表示你需要支付的最小费用。

动态规划，由于要花费最少，因此肯定是只吃n颗糖，且每天都至少要吃一颗，第i天至少需要吃i颗糖。

dp[i][j]代表在第i天时，总共购买j颗糖果需要的最少钱

#include<bits/stdc++.h>

using namespace std;

typedef pair<int,int>PII;
typedef long long ll;
const int N=310;
const int INF=0x3f3f3f3f;

int dp[N][N],a[N][N],sum[N][N];

void solve()
{
    int n,m;
    cin>>n>>m;
    
   for(int i=1;i<=n;i++)
       for(int j=1;j<=n;j++)
           dp[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];
    }
    
    //第一天需要特殊处理一下
    for(int i=1;i<=m;i++)
        dp[1][i]=sum[1][i]+i*i;
    
    for(int i=2;i<=n;i++)
        for(int j=i;j<=n;j++)//只需要算到第n天即可
            for(int k=i-1;k<=j;k++)//第i-1天购买的总糖果数
                if(j-k<=m)
                    dp[i][j]=min(dp[i][j],dp[i-1][k]+sum[i][j-k]+(j-k)*(j-k));
//     for(int i=1;i<=n;i++)
//     {
//         for(int j=1;j<=n;j++)
//             cout<<dp[i][j]<<" ";
//         cout<<'\n';
//     }
    cout<<dp[n][n]<<'\n';
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    
    int _=1;
    //cin>>_;
    while(_--)
        solve();
}