洛谷P1659 养猪

洛谷P1659 养猪 

#include <bits/stdc++.h> 
#define For(i,j,k) for(int i=j;i<=k;i++) 
using namespace std ; 

const int N = 1011 ; 
int n,day ; 
int f[N] ; 
struct node{
    int p,a ; 
}t[N];
inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline bool cmp(node a,node b) {
    if(a.p!=b.p) return a.p > b.p ; 
    return a.a > b.a ; 
}

int main()
{
    n = read() ; day = read() ; 
    For(i,1,n) t[i].a=read() ; 
    For(i,1,n) t[i].p=read() ; 
    sort(t+1,t+n+1,cmp) ; 
    For(i,1,n) 
      for(int j=max(i,day);j>=1;j--) 
        f[j] = max(f[j],f[j-1]+max(0,t[i].a-t[i].p*(j-1)) ) ; 
    int ans = -100 ; 
    For(i,1,day) ans = max(f[i],ans) ;      
    //  为什么最后还要再扫一遍，是因为有可能中间几个体重已经减没了，
    //  然后体重为0的猪也被当成了一头猪在计算 
    printf("%d
",ans) ; 
    return 0 ; 
}