牛客练习赛128：Cidoai的平均数对（背包dp）

题目描述

给定 nnn 对数 (ai,bi)(a_i,b_i)(ai​,bi​) 和参数 kkk，你需要选出一些对使得在满足 bib_ibi​ 的平均值不超过 kkk 的同时，aia_iai​ 的和最大，求出这个最大值。

输入描述:

第一行两个整数分别表示 n,kn,kn,k。
接下来 nnn 行，每行两个数分别表示 ai,bia_i,b_iai​,bi​

输出描述:

一行一个整数表示答案。

示例1

输入

复制5 6 4 10 3 4 6 7 7 7 10 8

5 6
4 10
3 4
6 7
7 7
10 8

输出

复制16

16

备注:

0≤ai,bi,k≤500,1≤n≤5000 \le a_i,b_i,k \le 500,1 \le n \le 5000≤ai​,bi​,k≤500,1≤n≤500

做法

本题重点在这个平均数的处理。b1+b2+b3+……+bn<=n*k，也就是(b1-k)+(b2-k)+(b3-k)+……+(bn-k)<=0。那我们就先把bi全都减去k。那bi为负数的就可以全部拿下。这样一来，我们背包的容量就是bi为负数的总和的绝对值了。

#include<bits/stdc++.h>
using namespace std;
const int N=510,M=250010;
int n,k;
int a[N],b[N];
int dp[M];
int res,ans,sum,ans2;
struct ty{
	int a,b;
};
vector<ty> v;
int main(){

    scanf("%d%d",&n,&k);
    v.push_back({-1,-1});

    for(int i=1;i<=n;i++) {
        cin>>a[i]>>b[i];
        b[i]-=k;
        if(b[i]<=0) {
			ans+=a[i];
			sum+=-b[i];
		}
		else{
			v.push_back({a[i],b[i]});
		}
    }
    
    memset(dp,-0x3f,sizeof(dp));
    dp[0]=0;
    
    for(int i=1;i<v.size();i++){
        for(int j=sum;j>=0;j--){   
            if(j-v[i].b>=0)
                dp[j]=max(dp[j],dp[j-v[i].b]+v[i].a);
        }
    }
    
    for(int i=0;i<=sum;i++) ans2=max(dp[i],ans2);

    cout<<ans+ans2;
}