[洛谷P1417 烹调方案]贪心+dp

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3211Dream City

---

Time Limit: 1 Second      Memory Limit: 32768 KB

---

JAVAMAN is visiting Dream City and he sees a yard of gold coin trees. There are n trees in the yard. Let's call them tree 1, tree 2 ...and tree n. At the first day, each tree i has a_i coins on it (i=1, 2, 3...n). Surprisingly, each tree i can grow b_i new coins each day if it is not cut down. From the first day, JAVAMAN can choose to cut down one tree each day to get all the coins on it. Since he can stay in the Dream City for at most m days, he can cut down at most m trees in all and if he decides not to cut one day, he cannot cut any trees later. (In other words, he can only cut down trees for consecutive m or less days from the first day!)

Given n, m, a_i and b_i (i=1, 2, 3...n), calculate the maximum number of gold coins JAVAMAN can get.

Input

There are multiple test cases. The first line of input contains an integer T (T <= 200) indicates the number of test cases. Then T test cases follow.

Each test case contains 3 lines: The first line of each test case contains 2 positive integers n and m (0 < m <= n <= 250) separated by a space. The second line of each test case contains npositive integers separated by a space, indicating a_i. (0 < a_i <= 100, i=1, 2, 3...n) The third line of each test case also contains n positive integers separated by a space, indicating b_i. (0 < b_i<= 100, i=1, 2, 3...n)

Output

For each test case, output the result in a single line.

Sample Input

2
2 1
10 10
1 1
2 2
8 10
2 3

Sample Output

10
21

Hints:
Test case 1: JAVAMAN just cut tree 1 to get 10 gold coins at the first day.
Test case 2: JAVAMAN cut tree 1 at the first day and tree 2 at the second day to get 8 + 10 + 3 = 21 gold coins in all

题意：你可以在m天每天砍掉一棵树，在第Q天砍掉的树w获得的价值=a[w]+b[w]*(Q-1),求能获得的最大价值。

题解：容易想到转移方程为dp[i][j]=max(dp[i][j],dp[i-1][j-1]+p[i].b*(j-1)+p[i].a);，但是由于转移方程中有一个变量的贡献和天数有关即p[i].b，所以可以贪心地想到应该让p[i].b大的排在后面进行更新，所以排序之后进行dp即可

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=1e4+5;
ll dp[255][255];
struct pot{
    ll a;
    ll b;
}p[255];
bool cmp(struct pot aa,struct pot bb){
    return aa.b<bb.b;
}
int main(){
    int t;
    cin>>t;
    while(t--){
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)scanf("%lld",&p[i].a);
        for(int i=1;i<=n;i++)scanf("%lld",&p[i].b);
        memset(dp,0,sizeof(dp));
        sort(p+1,p+1+n,cmp);
        ll ans=0;
        for(int i=1;i<=n;i++){
           for(int j=1;j<=m;j++){
                dp[i][j]=dp[i-1][j];
                dp[i][j]=max(dp[i][j],dp[i-1][j-1]+p[i].b*(j-1)+p[i].a);
           }
        }
        printf("%lld
",dp[n][m]);
    }
    return 0;
}

https://www.luogu.org/problemnew/show/P1417

一共有n件食材，每件食材有三个属性，ai，bi和ci，如果在t时刻完成第i样食材则得到ai-t*bi的美味指数，用第i件食材做饭要花去ci的时间。

众所周知，gw的厨艺不怎么样，所以他需要你设计烹调方案使得美味指数最大

输入格式：

第一行是两个正整数T和n，表示到达地球所需时间和食材个数。

下面一行n个整数，ai

下面一行n个整数，bi

下面一行n个整数，ci

输出格式：

输出最大美味指数

输入样例#1： 

74 1
502
2
47

输出样例#1：

408

【数据范围】

对于40%的数据1<=n<=10

对于100%的数据1<=n<=50

所有数字均小于100,000

题解：容易想到转移方程为dp[j]=max(dp[j],dp[j-p[i].c]+p[i].a-j*p[i].b);其中两个变量的贡献和天数有关即p[i].b,p[i].c，两种食材设为A,B 则他们排列为AB时的贡献为-(p[A].c+p[B].c)*p[B].b-p[A].c*p[A].b,他们排列为BA时的贡献为-(p[A].c+p[B].c)*p[A].b-p[B].c*p[B].b，当(p[A].c+p[B].c)*p[B].b-p[A].c*p[A].b<-(p[A].c+p[B].c)*p[A].b-p[B].c*p[B].b时，排列为AB比BA更优，化简之后就是p[A].c*p[B].b<p[A].b*p[B].c，所以可以以此排序，然后dp

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
ll dp[100005];
struct pot{
    ll a;
    ll b;
    ll c;
}p[55];
bool cmp(struct pot ab,struct pot ba){
    return (ab.c*ba.b<ab.b*ba.c);
}
int main(){
    int t,n;
    scanf("%d%d",&t,&n);
    for(int i=1;i<=n;i++)scanf("%lld",&p[i].a);
    for(int i=1;i<=n;i++)scanf("%lld",&p[i].b);
    for(int i=1;i<=n;i++)scanf("%lld",&p[i].c);
    ll ans=0;
    sort(p+1,p+1+n,cmp);
    for(int i=1;i<=n;i++){
        for(int j=t;j>=p[i].c;j--){
            //dp[j]=max(dp[j],dp[j-1]);
            dp[j]=max(dp[j],dp[j-p[i].c]+p[i].a-j*p[i].b);
            ans=max(ans,dp[j]);
        }
    }
    cout<<ans<<endl;
    return 0;
}