题目:
这道题只要出现了双重循环,不管如何进行剪枝都会导致TLE。只能使用其他的方法。
AC代码(附讲解):
首先介绍两个函数:
- lower_bound 是经常用来对STL容器进行查找的函数,其前两个参数是容器的迭代器类型,分别是,查找范围的 begin() 迭代器 和 查找范围的 end() 迭代器 第三个参数是待查找的数值 num 函数返回的结果是容器中第一个大于等于num的值得下标。
- upper_bonud 参数如上,函数返回的是第一个 大于 num 的值的下标
Tips:
1. 这两种查找的方式都是采用的二分查找 (时间复杂度较低)
2. 如果不存在符合条件的值就会直接返回容器的end()迭代器
#include<bits/stdc++.h>
#define ll long long
using namespace std;
vector<int> num;
int main(void)
{
int n,x,y;
cin >> n >> x >> y;
for(int i = 0;i < n;i ++) // 尽量不要直接对vector的容器进行按照下标进行读入的操作,读取可以随机查下标
{
int x;
cin >> x;
num.push_back(x);
}
sort(num.begin(),num.end()); // 先排序,这道题里面数字的排序并不会影响组合的种数(排序也是二分查找的前提)
ll ans = 0;
for(int i = 0;i < num.size();i ++)
{
// 注意查找的范围是在当前元素的下一个到容器的末尾
int pos1 = lower_bound(num.begin() + i + 1,num.end(),x - num[i]) - num.begin();
// x - num[i] 表示对当前的元素而言需要满足下限所需要的相加的最小的元素 即 可选择的加数范围的左边 (可以取到这个下标)
int pos2 = upper_bound(num.begin() + i + 1,num.end(),y - num[i]) - num.begin();
//y - num[i] 则表示对当前的元素而言需要满足上线所需要的最大的元素 即 可选择的加数的范围的右边(但不可以取到这个下标)
// 综上所述,对于当前所遍历到的每一个元素 其可选加数的范围就是[pos1,pos2)
// Tips: 作图更加方便理解
ans += (pos2 - pos1);
}
cout << ans;
return 0;
}