蓝桥杯--枚举1 AcWing 1236. 递增三元组

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);
		
	}
}

---

看下两边的代码真是差多了，不过能过就行啊哈哈