蓝桥杯第九届省赛第6题
标题:递增三元组
给定三个整数数组
A = [A1, A2, … AN],
B = [B1, B2, … BN],
C = [C1, C2, … CN],
请你统计有多少个三元组(i, j, k) 满足:
- 1 <= i, j, k <= N
- Ai < Bj < Ck
【输入格式】
第一行包含一个整数N。
第二行包含N个整数A1, A2, … AN。
第三行包含N个整数B1, B2, … BN。
第四行包含N个整数C1, C2, … CN。
对于30%的数据,1 <= N <= 100
对于60%的数据,1 <= N <= 1000
对于100%的数据,1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000
【输出格式】
一个整数表示答案
【输入样例】
3
1 1 1
2 2 2
3 3 3
【输出样例】
27
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 1000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
不要使用package语句。不要使用jdk1.7及以上版本的特性。
主类的名字必须是:Main,否则按无效代码处理。
思路:
这题最容易想到的就是直接暴力,三重for循环进行判断,这个应该大家都写的出,但是数据规模大肯定会超时,这里需要用到二分中的lower_bound和upper_bound函数
我们可以以B数组作为中间量,判断在A数组中小于B[i]的元素有多少,判断在C数组中大于B[i]的元素有多少。
代码:
public class Main6 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[] a = new int[n];
int[] b = new int[n];
int[] c = new int[n];
for (int i = 0; i < n; i++) {
a[i] = sc.nextInt();
}
for (int i = 0; i < n; i++) {
b[i] = sc.nextInt();
}
for (int i = 0; i < n; i++) {
c[i] = sc.nextInt();
}
Arrays.sort(a);
Arrays.sort(b);
Arrays.sort(c);
int sum = 0;
int ares;
int cres;
for (int i = 0; i < n; i++) {
ares = lower_bound(a, 0, n, b[i]);
cres = n - upper_bound(c, 0, n, b[i]);
sum += (ares * cres);
}
System.out.println(sum);
}
//求小于等于key的序号
private static int upper_bound(int[] a, int start, int end, int key) {
while (start <= end) {
int mid = start + (end - start) / 2;
if (a[mid] < key) {
start = mid + 1;
} else {
end = mid;
}
}
return start;
}
//小于key的序号
private static int lower_bound(int[] a, int start, int end, int key) {
while (start < end) {
int mid = start + (end - start) / 2;
if (a[mid] < key) {
start = mid + 1;
} else {
end = mid;
}
}
return start;
}
}