【二分算法】组建队伍

       题目如下。本题其实不难，但因要求每步执行不超过1000ms，所以不能套用多重循环，否则会Time Exceeded，所以只能优化算法。百度搜索对于3个数据排序比较类型题算法很少，分享思路如下：

      1）3个班录入数据后进行排序；

      2）只针对2班的同学身高进行遍历，搜索b[i]分别在1班和3班的位置并保存数据；

      3）在1班中，b[i]之前的人数和在3班中b[i]之后的人数是符合要求的，结果相乘即可（这是关键思路），然后累加每个b[i]的乘积即为答案！

注意：需要考虑身高相同的情况。尤其是3班中有可能连续几个相同身高且都刚好等于b[i]的情况，所以不能用 C++ 标准库中的lower_bound函数，自建一个二分查找函数找到第一个大于b[i]的位置即可。同时需关注题目给的数据范围，做好变量类型声明。

附DEV C++源码及详细说明，测试AC通过。

#include <bits/stdc++.h>
using namespace std;
//二分查找：找出第一个大于x的元素位置,从下标0开始
int findFirstGreater(int arr[], int size, int x) { 
	int low = 0;
	int high = size - 1;
	int result = -1; // 如果找不到符合条件的元素，返回-1
	while (low <= high) {
		int mid = low + (high - low) / 2;
		if (arr[mid] > x) {
			// 如果中间元素大于x，检查是否是第一个大于x的元素
			if (mid == 0 || arr[mid - 1] <= x) {
				result = mid; // 是第一个大于x的元素，或者前一个元素不大于x
				high = mid - 1; // 继续在左半部分搜索，可能还有更小的元素也大于x
			} else {
				high = mid - 1; // 前一个元素大于x，继续在左半部分搜索
			}
		} else {
			low = mid + 1; // 中间元素不大于x，继续在右半部分搜索
		}
	}
	return result;
}

int main() { //组建队伍
	int n, ans1 = 0, ans2 = 0;
	long long count = 0;
	cin >> n;
	int a[n ], b[n], c[n ];
	for (int i = 0; i < n; i++) cin >> a[i];
	for (int i = 0; i < n; i++) cin >> b[i];
	for (int i = 0; i < n; i++) cin >> c[i];
	sort(a, a + n ); //升序排序3个班
	sort(b, b + n);
	sort(c, c + n);
	for (int i = 0; i < n; i++) {
		auto pos1 = lower_bound(a, a + n, b[i]);//找到第一个大于或等于的元素位置
		if (pos1 != a + n) { //如果找到
			ans1 = distance(a, pos1);//从a班中找到符合的人数
		} else { //没有找到
			ans1 = n;//全部人数符合
		}
		int pos2 = findFirstGreater(c, n, b[i]); //找到第一个大于b[i]的元素位置
		if (pos2 != -1) { //如果找到
			ans2 = n - pos2; //从c班中找出符合的人数
		} else //没有找到
			ans2 = 0;
		count += (long long)ans1 * ans2;//根据题目数据范围，此处必须要加long long，否则会出错
	}
	cout << count;
	return 0;
}