CSUOJ1630--Plane Ticket Pricing

类似于背包,但是最后物品可以拆分.另外,因为要求第一次的选择,从后往前进行dp.

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

#define INF 0x3f3f3f3f

typedef struct
{
     int cnt;
     int w[103],v[103];
}P;
P p[55];

int dp[55][550];

int main()
{
      //freopen("in.txt","r",stdin);
       int n,m;
       while(scanf("%d%d",&n,&m)!=EOF)
       {
              m++;
              memset(dp,0,sizeof(dp));
              for(int i=m;i>=1;i--)
              {
                    scanf("%d",&p[i].cnt);
                    for(int j=0;j<p[i].cnt;j++)
                         scanf("%d",&p[i].w[j]);
                    for(int j=0;j<p[i].cnt;j++)
                         scanf("%d",&p[i].v[j]);
              }
              memset(dp,-1,sizeof(dp));
              dp[0][0]=0;
              for(int i=1;i<=m;i++)
              {
                  for(int j=n;j>=0;j--)
                  {
                         for(int k=0;k<p[i].cnt;k++)
                         {
                               if(j-p[i].v[k]>=0&&dp[i-1][j-p[i].v[k]]!=-1)
                               {
                                      dp[i][j]=max(dp[i][j],dp[i-1][j-p[i].v[k]]+p[i].w[k]*p[i].v[k]);
                               }
                                if(j==p[i].v[k]) ///可以卖任意多张的票
                                      for(int d=p[i].v[k]-1;d>=0;d--)
                                          if(dp[i-1][0]!=-1)
                                               dp[i][d]=max(dp[i][d],dp[i-1][0]+d*p[i].w[k]);
                          }
                  }
              }
        int sum=0;
        for(int i=0;i<=n;i++)
            sum=max(dp[m][i],sum);
        if(m==1) ///m==1的情况需特判
        {
            printf("%d\n",sum);
            for(int i=0;i<p[m].cnt;i++)
                if(p[m].w[i]*p[m].v[i]==sum)
                {
                      printf("%d\n",p[m].w[i]);
                      break;
                }
            continue;
        }
        printf("%d\n",dp[m][n]);
        for(int i=0;i<p[m].cnt;i++)
              if(dp[m][n]==dp[m-1][n-p[m].v[i]]+p[m].v[i]*p[m].w[i])
              {
                    printf("%d\n",p[m].w[i]);
                    break;
              }
       }
   return 0;
}