原题连接:上海市计算机学会竞赛平台 | YACS
一,70分的写法
先对能力值进行从小到大排序,在对热情度进行前缀和处理,再枚举区间中的开始与结束位置判断是否满足能力值最大选手的能力差值小于等于X的条件,如果满足则利用前面的前缀和使ans = ans和[i,j]区间内的热情度之和取最大值 ,最后输出ans即可。
时间复杂度:O(N^2)
代码:
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
using namespace std;
struct xs
{
long long a,b;
} s[1000001];
long long n,ans,x;
bool cmp(xs a1,xs a2)
{
return a1.a < a2.a;
}
int main()
{
scanf("%lld",&n);
for(int i = 1; i <= n; i++) scanf("%lld%lld",&s[i].a,&s[i].b);
scanf("%lld",&x);
sort(s + 1,s + 1 + n,cmp);//按能力值从小到大排序
for(int i = 2; i <= n; i++) s[i].b += s[i - 1].b;//为了等下可以O(1)求出[i,j]区间内的热情度之和
for(int i = 1; i <= n; i++)//枚举区间中的起始位置
{
if(ans > s[n].b - s[i - 1].b) break;//如果目前的答案>[i - 1,n]区间内的热情之和,说明后续的枚举是没有意义的,永远不会超过ans,跳出循环
for(int j = i + 1; j <= n; j++)//枚举区间中的结束位置
if(s[j].a - s[i].a <= x)//满足条件
{
if((s[j].b - s[i - 1].b) > ans)//ans和[i,j]区间内的热情度之和取最大值
ans = s[j].b - s[i - 1].b;
}
else break;//如果不满足条件,那么后续枚举结束位置的操作只会增加[i - 1,n]区间内的热情之和的值,所以直接跳出循环
}
printf("%lld",ans);
return 0;
}
二,100分
用二分查找来查找结束位置
时间复杂度:O(log n)
代码:
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
using namespace std;
struct xs
{
long long a,b;
} s[1000001];
long long n,ans,x;
bool cmp(xs a1,xs a2)
{
return a1.a < a2.a;
}
int main()
{
scanf("%lld",&n);
for(int i = 1; i <= n; i++) scanf("%lld%lld",&s[i].a,&s[i].b);
scanf("%lld",&x);
sort(s + 1,s + 1 + n,cmp);//按能力值从小到大排序
for(int i = 2; i <= n; i++) s[i].b += s[i - 1].b;//为了等下可以O(1)求出[i,j]区间内的热情度之和
for(int i = 1; i <= n; i++)//枚举区间中的起始位置
{
if(ans > s[n].b - s[i - 1].b) break;//如果目前的答案>[i - 1,n]区间内的热情之和,说明后续的枚举是没有意义的,永远不会超过ans,跳出循环
int left = i + 1,right = n;
while(left <= right)
{
int mid = (left + right) / 2;
if(s[mid].a - s[i].a > x) right = mid - 1;
else left = mid + 1;
}
ans = max(ans,s[right + left >> 1].b - s[i - 1].b);
}
printf("%lld",ans);
return 0;
}