AcWing 1236. 递增三元组
给定三个整数数组
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。
输出格式
一个整数表示答案。
数据范围
1≤N≤105,
0≤Ai,Bi,Ci≤105
输入样例:
3
1 1 1
2 2 2
3 3 3
输出样例:
27
题意:就找出有多少递增三元组就行了
其实y总的做法是更简洁的,这里还是分享一下我的笨比做法,倒是也能通过所有数据
也是看到二分和前缀和的标签想到了这种做法,首先对三元组升序排序,对于b组的每个元素b[i],找出a组最大的小于b[i]的个数,也就是下标,存在rb数组中,再同理处理b组和c组
这里结束后是可以用算式处理的,不过我用的思想是当找出b组中小于c[i]的个数kl后,说明b组中前k个元素都是小于c[i]的,再结合a组中小于b[i]的个数,将前k个rb相加求和即可
这里为了简化时间,对rb进行前缀和处理
代码如下
import java.io.*;
import java.util.*;
public class Main {
static Scanner tab = new Scanner(System.in);
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
static int N = 100010;
static int a[]=new int [N];
static int b[]=new int [N];
static int c[]=new int [N];
//求出a组中最大小于b[i]的元素个数即坐标
public static int finda(int l,int r,int k) {
if(l==r) {
if(a[l]<k)
return l;
else
return 0;
}
int mid=(l+r+1)/2;//寻找最大小于值做取右处理
if(a[mid]<k) {
return finda(mid,r,k);
}
else {
return finda(l,mid-1,k);
}
}
//求出b组中最大小于c[i]的元素个数即坐标
public static int findb(int l,int r,int k) {
if(l==r) {
if(b[l]<k)
return l;
else
return 0;
}
int mid=(l+r+1)/2;//同理
if(b[mid]<k) {
return findb(mid,r,k);
}
else {
return findb(l,mid-1,k);
}
}
public static void main(String[] args) throws IOException {
int n=tab.nextInt();
//记录个数并做前缀和处理,开int会爆掉
long rb[]=new long [N];
for(int i=1;i<=n;i++) {
a[i]=tab.nextInt();
}
for(int i=1;i<=n;i++) {
b[i]=tab.nextInt();
}
for(int i=1;i<=n;i++) {
c[i]=tab.nextInt();
}
Arrays.sort(a,1,n+1);
Arrays.sort(b,1,n+1);
Arrays.sort(c,1,n+1);
for(int i=1;i<=n;i++) {
rb[i]=finda(1, n, b[i]);
}
//前缀和处理
for(int i=1;i<=n;i++) {
rb[i]+=rb[i-1];
}
long res=0;//同理开long
for(int i=1;i<=n;i++) {
int x=findb(1,n,c[i]);
res+=rb[x];
}
System.out.println(res);
}
}
看下两边的代码真是差多了,不过能过就行啊哈哈