第九届蓝桥杯省赛C++B组 递增三元组

标题:递增三元组
给定三个整数数组
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

资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗  < 1000ms

请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。
注意:
main函数需要返回0;
只使用ANSI C/ANSI C++ 标准;
不要调用依赖于编译环境或操作系统的特殊函数。
所有依赖的函数必须明确地在源文件中 #include <xxx>
不能通过工程设置而省略常用头文件。
提交程序时,注意选择所期望的语言类型和编译器类型。


思路:因为要求是递增序列,那么分别把ABC三个数组各自排序,这里我用了一个upper_bound函数(返回第一个大于所查询值的地址),所以只用了二重循环,二重循环里面,我又用了一个temp来记录上一次B[j]>A[i]开始的下标,稍微减少了一些循环次数。

代码:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<map>  
#include<set>
using namespace std;

#define N 100005
int A[N],B[N],C[N];

int main()
{
	int n;
	cin>>n;
	for(int i=0;i<n;i++)
		cin>>A[i];
	for(int i=0;i<n;i++)
		cin>>B[i];
	for(int i=0;i<n;i++)
		cin>>C[i];
	sort(A,A+n);
	sort(B,B+n);
	sort(C,C+n);
	int flag=0,temp=0;
	long long ans=0;
	for(int i=0;i<n;i++)
	{		
		for(int j=temp;j<n;j++)
		{
			if(B[j]>A[i])
			{
				if(!flag)
				{
					temp=j;
					flag=1;
				}
				ans+=C+n-upper_bound(C,C+n,B[j]);
			}
		}
		flag=0;
	}
	cout<<ans<<endl;	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/ryo_218/article/details/79822620