CF#321-B - Kefa and Company 二分+前缀和

时间:2022-12-30 00:07:43

传送门:

题目大意

Kefa 有n个朋友,她请客,为了让所有的朋友不感到彼此之间的贫穷,她请客的朋友的钱数最小值和最大值之间不能超过d。
输入的时候输入n和d
下面的n行代表n个朋友,第一个数代表这个朋友的钱数,第二个整数代表Kefa 和他的友情值,最后输出在最大值和最小值不超过d的情况下获得友谊的最大值。

解题思路

我们先来对朋友所拥有的钱数按照升序排列,sum[i]表示前i个朋友的友谊值总和。
然后查找从钱最少的开始循环,依次找到大于他不超过d的友谊值总和。取出最大的即可。

AC代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std ;
const int MAXN = 1e5+5;
long long sum[MAXN];
long long tmpM[MAXN];

struct Node{
long long money;
long long ship;
}friends[MAXN<<2];

bool cmp(const Node a,const Node b)
{
return a.money<b.money;
}

int main()
{
int n,d;
scanf("%d%d",&n,&d);
for(int i=0;i<n;i++)
scanf("%lld%lld",&friends[i].money,&friends[i].ship);
sort(friends,friends+n,cmp);
for(int i=1;i<=n;i++)
{
tmpM[i-1] = friends[i-1].money;
sum[i] += sum[i-1] + friends[i-1].ship;
}
long long ans = -1;
for(int i=0;i<n;i++)
{
int p=upper_bound(tmpM,tmpM+n,friends[i].money+d-1)-tmpM;
ans = max(ans,sum[p]-sum[i]);
}
printf("%lld\n",ans);
return 0;
}