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

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

#include
#include 
#include 
#include 
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;
}

你可能感兴趣的:(DP)