【蓝桥杯】 2018初赛 递增三元组

题目描述

给定三个整数数组
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。
1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000

输出

一个整数表示答案

样例输入
3
1 1 1
2 2 2
3 3 3
样例输出
27

【题解】：
1.暴力可解，但是超时pass。
改进暴力：遍历b数组，用b数组确定a满足条件的有多少个，c数组确定满足条件的有多少个，然后相乘。再相加。

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int a[100005];
int b[100005];
int c[100005];

int main()
{
    int n;
    scanf("%d",&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 aa=0;//a数组标记
    ll cc=0;//c数组标记
    ll ans=0;
    for(int i=0;i<n;i++)
    {
       // cout<<"----ai="<<aa<<" cc="<<cc<<endl;
        while(a[aa]<b[i]&&aa<n)//注意条件，不能等于
            aa++;
        while(c[cc]<=b[i]&&cc<n)//同上
            cc++;
      // cout<<"ai="<<aa<<" cc="<<cc<<endl;
        ans+=aa*(n-cc);
    }
    cout<<ans<<endl;
}