购物(DP&贪心)

购物(DP&贪心)

思路： $dp+$贪心。

令 $dp[i][j]$为前 $i$天买 $j$个糖果的最小花费，然后贪心的选取每天价格最小的前 $k$个， $dp$转移即可。此外注意下一些条件的限制。

时间复杂度: $O(n^3)$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=350+5,M=2e4+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a,b) memset(a,b,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair<int,int>
#define fi first
#define se second
#define pb push_back
int dp[N][N],n,m,pre[N][N],a[N];
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++)
            scanf("%d",&a[j]);
        sort(a+1,a+m+1);
        for(int j=1;j<=m;j++)
          pre[i][j]=pre[i][j-1]+a[j];
    }
    mst(dp,0x3f);
    dp[0][0]=0;
    for(int i=1;i<=n;i++){
        for(int j=i;j<=min(n,i*m);j++)
          for(int k=i-1;k<=min({n,(i-1)*m,j});k++){
           dp[i][j]=min(dp[i][j],dp[i-1][k]+pre[i][j-k]+(j-k)*(j-k));
          }
    }
    printf("%d\n",dp[n][n]);
	return 0;
}