本题是阿里蚂蚁金服面试题:
给定一个数组s,和一个数a,在数组s里面找到3个数使这3个数的和小于a,即s[i]+s[j]+s[k]<a,请你设计一个函数返回数组s里这样的组合有多少个?
int f(double s[], double a) {
int cnt = 0;
Arrays.sort(s);
int i = 0;
int j = s.length;
while (i < j - 1) {
int low = i;
int high = j;
while (low < high) {
int k = (low + high) / 2;
if (s[i] + s[j] + s[k] >= a) {
high = k - 1;
} else if (s[i] + s[j] + s[k] < a) {
// 下面区间都是满足的
cnt += k - low;
// 二分查找
low = k + 1;
}
}
// 扫描结束,根据情况移动上下限
if (s[i] + s[j] > a) {
j--;
} else {
i++;
}
}
return cnt;
}