P1102 A-B 数对(二分 STL函数的运用)
思路:
查找A-B = C的对数,则A = B+C,等价于枚举B,查找,数组中有多少个等于B+C的个数,再累加起来
upper_bound():返回的是被查序列中第一个大于查找值得位置;
lower_bound():返回的是被查序列中第一个大于等于查找值的位置
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 2e5+7;
LL num[maxn];
LL cnt = 0;
LL n,c;
int main()
{
cin>>n>>c;
for(int i=1;i<=n;i++)
{
cin>>num[i];
}
sort(num+1,num+n+1);
//枚举B
LL ans = 0;
for(LL i=1;i<=n;i++)
{
LL add = num[i] + c;
LL pos = lower_bound(num+1,num+n+1,add)-num;//返回的是数组中第一个大于等于add的位置
if(pos==n+1) continue;//没有数值等于B+C的情况
else
{
LL p = upper_bound(num+1,num+n+1,add)-num;//返回大于add的下标
ans += p - pos;
}
}
printf("%lld\n",ans);
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话