给定三个整数数组 A = [ A 1 , A 2 , ⋯ , A N ] A = [A_1, A_2,\cdots, A_N] A=[A1,A2,⋯,AN], B = [ B 1 , B 2 , ⋯ , B N ] B = [B_1, B_2,\cdots, B_N] B=[B1,B2,⋯,BN], C = [ C 1 , C 2 , ⋯ , C N ] C = [C_1, C_2,\cdots,C_N] C=[C1,C2,⋯,CN]。
请你统计有多少个三元组 ( i , j , k ) (i, j, k) (i,j,k) 满足:
第一行包含一个整数 N N N。
第二行包含 N N N 个整数 $ A_1, A_2,\cdots, A_N$。
第三行包含 N N N 个整数 $ B_1, B_2,\cdots, B_N$。
第四行包含 N N N 个整数 $ C_1, C_2,\cdots, C_N$。
一个整数表示答案
3
1 1 1
2 2 2
3 3 3
27
对于 30 % 30\% 30% 的数据, 1 ≤ N ≤ 100 1 \le N \le 100 1≤N≤100。
对于 60 % 60\% 60% 的数据, 1 ≤ N ≤ 1000 1 \le N \le 1000 1≤N≤1000。
对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 1 0 5 1 \le N \le 10^5 1≤N≤105, 0 ≤ A i , B i , C i ≤ 1 0 5 0 \le A_i, B_i, C_i \le 10^5 0≤Ai,Bi,Ci≤105。
所需变量
int n;//代表每个元组有多少个数字int arr[3][100005] = {0};//输入进来用于存储每个数
int i,j,k;//循环变量
int temp;//用于接收每个输入的数
int barr[3] = {0};//用于存储目前有多少个数满足
long long sum = 0;//代表最后有多少满足的
思路:我们可以暴力遍历每个数组,当arr[0][i]选取一个数之后,我们只能在这个数后面不断遍历是否存在,如果存在就再最后满足的三元组里面+1,不断遍历结束,我们就的到最后答案了!这样做确实可以,不过在时间复杂度上是O(n^3)
,这个时间复杂度是不可接受的,所以我们的做法是将a数组每次比他小的数我们都将数量累加到最大的那个数上!这样解释可能不太理解!
我们举个例子也就是目前a数组有以下几个数1 2 3 4 5 6 7 8 9
我们的做法是将这几个数当初的做法就是以这几个数为下标的数组都赋值为1,那么我们得到的数组arr[0][10]中存储的值都是1,我们现在不想遍历那么多次了,我们优先遍历一遍,对于3这个数来说,有1,2两个数都比他小,如果a数组能选3组成三元组,那么1和2肯定也是可以的,所以我们将arr[0][3]存储的数赋值为3,那么我们就不需要往下遍历了,我们就将O(n^3)
的时间复杂度降为O(n^2)
因此将后面每个数存在并且此做法的代码如下:
for(i = 0;i<100005;i++){
if(arr[0][i] != 0){
barr[0] += arr[0][i];
arr[0][i] = barr[0];
}
得到a数组的优化后,我们从c数组最大往下遍历,每次b遍历到最小,遍历过程中,只要找到a数组中第一个比b数组小的,我们就知道比a数组这个数小的还有多少个,因此可以直接累乘起来!最后将最后的答案相累加得到做中的答案!
for(i = 100005;i>=2;i--){
if(arr[2][i] == 0){
continue;
}
for(j = i-1;j>=1;j--){
if(arr[1][j] == 0){
continue;
}
for(k = j-1;k>=0;k--){
if(arr[0][k] == 0){
continue;
}
sum +=arr[2][i]*arr[1][j]*arr[0][k];
k = -1;
}
}
}
后面那里直接将k赋值为-1代表已经找到这个a数组了,不需要往下遍历了,可以直接去找下一个满足条件的三元组!
完整代码如下(编译器是dev,语言是C语言):
#include<iostream>
using namespace std;
int main(){
int n,arr[3][100005] = {0},i,j,k,temp,barr[3] = {0};
int max = 0,min = 0;
long long sum = 0;
cin>>n;
for(i = 0;i<3;i++){
for(j = 0;j<n;j++){
cin>>temp;
arr[i][temp]++;
}
}
for(i = 0;i<100005;i++){
if(arr[0][i] != 0){
barr[0] += arr[0][i];
arr[0][i] = barr[0];
}
for(i = 100005;i>=2;i--){
if(arr[2][i] == 0){
continue;
}
for(j = i-1;j>=1;j--){
if(arr[1][j] == 0){
continue;
}
for(k = j-1;k>=0;k--){
if(arr[0][k] == 0){
continue;
}
sum +=arr[2][i]*arr[1][j]*arr[0][k];
k = -1;
}
}
}
cout<<sum;
return 0;
}