Dream City ZOJ - 3211

点击打开链接

dp[i][j]代表前i棵树在前j天的最大收益

dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]+tree[i].a+tree[i].b*(j-1)) 表示第i棵树砍还是不砍

这样写 首先要对n棵树按先b后a升序排序 因为这里用dp主要是因为a的因素 导致选哪些树不确定 一旦砍哪些树确定了 一定是b值大的在后砍

#include <bits/stdc++.h>
using namespace std;

struct node
{
    int a;
    int b;
};

node tree[300];
int dp[300][300];
int n,m;

bool cmp(node n1,node n2)
{
    if(n1.b==n2.b) return n1.a<n2.a;
    else return n1.b<n2.b;
}

int main()
{
    int t,i,j;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&tree[i].a);
        }
        for(i=1;i<=n;i++)
        {
            scanf("%d",&tree[i].b);
        }
        sort(tree+1,tree+n+1,cmp);
        memset(dp,0,sizeof(dp));
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=m;j++)
            {
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]+tree[i].a+tree[i].b*(j-1));
            }
        }
        printf("%d\n",dp[n][m]);
    }
    return 0;
}