给定三个整数数组
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
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
注意:
main函数需要返回0;
只使用ANSI C/ANSI C++ 标准;
不要调用依赖于编译环境或操作系统的特殊函数。
所有依赖的函数必须明确地在源文件中 #include
不能通过工程设置而省略常用头文件。
提交程序时,注意选择所期望的语言类型和编译器类型。
法一:由于这个和顺序没有关系,我们可以对这三个数组排序,然后枚举B数组的每一个值当做中间值,在A数组中找比他小的,在C中找小于等于的,然后与总数做差的大于的数,然后相乘即可。
#include <bits/stdc++.h>
using namespace std;
const int N = 100001;
int n;
int A[N], B[N], C[N];
long long Ans;
int Read();
int main() {
cin >> n;
for(int i=0; i<n; ++i) A[i] = Read();
for(int i=0; i<n; ++i) B[i] = Read();
for(int i=0; i<n; ++i) C[i] = Read();
sort(A, A+n);
sort(B, B+n);
sort(C, C+n);
for(int i=0, j=0, k=0; j<n; ++j) {
while(i<n && A[i]<B[j]) i++;
while(k<n && C[k]<=B[j]) k++;
Ans += (long long)i * (n-k);
}
printf("%lld\n", Ans);
return 0;
}
inline int Read() {
int x = 0, f = 1;
char ch = getchar();
while(ch<'0' || ch>'9') {
if(ch == '-') f = -1;
ch = getchar();
}
while(ch>='0' && ch<='9') {
x = x * 10 + ch - '0';
ch = getchar();
}
return x * f;
}
法二:先排序,以b[i]作为解题关键,利用upper_bound和lower_bound可以找到a[i]中小于b[i]的个数x,c[i]中大于b[i]的个数y,就是说以b[i]作为中间,b[i]左侧有x种选择,右侧有y种选择,所以以b[i]为中间序列的有x*y种,遍历一遍b[i]
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
int a[MAXN],b[MAXN],c[MAXN];
int n,sum;
int main()
{
cin>>n;
for(int i=0;i<n;i++)scanf("%d",&a[i]);
for(int i=0;i<n;i++)scanf("%d",&b[i]);
for(int i=0;i<n;i++)scanf("%d",&c[i]);
sort(a,a+n);
sort(b,b+n);
sort(c,c+n);
sum = 0;
for(int i=0;i<n;i++){
int x = (lower_bound(a,a+n,b[i]) - a);//lower_bound是a[i]大于等于b[i]的第一个数首地址
int y = (n - (upper_bound(c,c+n,b[i]) - c));//upper是c[i]大于b[i]的第一个数首地址
sum += x*y;
}
cout<<sum;
return 0;
}