标题:递增三元组
给定三个整数数组
A = [A1, A2, ... AN],
B = [B1, B2, ... BN],
C = [C1, C2, ... CN],
请你统计有多少个三元组(i, j, k) 满足:
1. 1 <= i, j, k <= N
2. 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
思路:二分,对于b数组的每个数,我们去a数组里找第一个比b[i]小的数的位置,去c数组里找第一个比b[i]大的数的位置,这样
2*Nlog(N)就能找到所有情况.
代码:
#include<bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
const double esp = 1e-7;
const int ff = 0x3f3f3f3f;
map<int,int>::iterator it;
int n;
int a[maxn],b[maxn],c[maxn];
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);
ll ans = 0;
for(int i = 0;i< n;i++)
{
int pos1 = lower_bound(a,a+n,b[i])-a;
int pos2 = upper_bound(c,c+n,b[i])-c;
ans+= (ll)pos1*(n-pos2);
}
cout<<ans<<endl;
return 0;
}