CodeChef Online Shopping

题目

简单dp,用dp[i][j]表示第i个商品在第j家商店买的最小值(前面的i-1件商品已经算进去了),显然dp[i][j]=Min(dp[i-1][k],0<=k<=m)+买i商品的消费.

因为n,m可能过大,所以用滚动数组.

显然在前一次的计算中可以记录下,买i-1件商品时的最优值,和在哪买会有最优值.所以在计算第i件商品的时候,可以直接利用前一次记录的最优值...

注意是M家店,n个商品.

注意虽然n,m<=10W,开二维数组肯定不行.....所以可以用Vector,或者像我这样SB的用动态申请........

#include<stdio.h>
#include<algorithm>
#include<stdlib.h>
#include<string.h>
using namespace std;

typedef long long ll;
#define INF ~0ULL>>1
int **pri,**dis;
int t,n,m,tag,ttag;
ll dp[2][100000+5];
ll Min,MMin,val,vval,vvval,now;

int main()
{
    //printf("%lld\n",INF);
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        pri=(int **)malloc(m*sizeof(int*));
        dis=(int **)malloc(m*sizeof(int*));
        for(int i=0;i<m;i++)
        {
            pri[i]=(int *)malloc(n*sizeof(int));
            dis[i]=(int *)malloc(n*sizeof(int));
        }

        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                scanf("%d",&pri[j][i]);
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                scanf("%d",&dis[j][i]);
        Min=INF;
        tag=-1;
        now=0;
        for(int i=0;i<m;i++)
        {
            dp[now][i]=pri[i][0]*1LL;
            if(dp[now][i]<Min)
            {
                Min=dp[now][i];
                tag=i;
            }
        }
        for(int i=1;i<n;i++)
        {
            now^=1;
            MMin=INF,ttag=-1;
            for(int j=0;j<m;j++)
            {
                if(tag==j)
                {
                    val=pri[j][i]*1LL-dis[j][i-1];
                    dp[now][j]=Min+(val<0?0:val);
                    if(MMin>dp[now][j])
                    {
                        MMin=dp[now][j];ttag=j;
                    }
                }
                else
                {
                    vvval=pri[j][i]*1LL-dis[j][i-1];
                    val=Min+pri[j][i];
                    vval=dp[1-now][j]+(vvval<0?0:vvval);
                    dp[now][j]=min(val,vval);
                    if(dp[now][j]<MMin)
                    {
                        MMin=dp[now][j];ttag=j;
                    }
                }
            }
            Min=MMin,tag=ttag;
        }
        ll ans=INF;
        for(int i=0;i<m;i++)
        {
            ans=min(ans,dp[now][i]);
        }
        printf("%lld\n",ans);

        for(int i=0;i<m;i++)
        {
            free(pri[i]),free(dis[i]);
        }
        free(pri),free(dis);
    }
    return 0;
}