二分查找 + AcWing 1236. 递增三元组

模板

bool check(int x) {/* ... */} // 检查x是否满足某种性质

// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用：
int bsearch_1(int l, int r)
{
    while (l < r)
    {
        int mid = l + r >> 1;
        if (check(mid)) r = mid;    // check()判断mid是否满足性质
        else l = mid + 1;
    }
    return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用：
int bsearch_2(int l, int r)
{
    while (l < r)
    {
        int mid = l + r + 1 >> 1;
        if (check(mid)) l = mid;
        else r = mid - 1;
    }
    return l;
}

区别: 看下一个合法区间在左半边还是右半边，即mid属于哪个区间，如果下一个合法区间在左半边，则r = mid;否则为l = mid; 前提条件：mid一定要是合法的！！！

题目描述

给定三个整数数组
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

题解

1. 对a , b, c 三个数组排序。
2. a对于b数组，找的是小于b[i]的最大值，因此l=mid；之后再l+1算出a的数量， 因为下标是从0开始
3. c对于a数组，找的是大于a[i]的最小值，因此r=mid；之后再n-r算出c的数量,因为下标是从0开始

代码

#include<bits/stdc++.h>
using namespace std;
const int N=110000;
int a[N],b[N],c[N];
int n;
long long int mul;
int main()
{
    cin>>n;
    for(int i=0;i<=n-1;i++)
    {
        cin>>a[i];
    }
    for(int i=0;i<=n-1;i++)
    {
        cin>>b[i];
    }
    for(int i=0;i<=n-1;i++)
    {
        cin>>c[i];
    }
    sort(a,a+n);
    sort(b,b+n);
    sort(c,c+n);
    
    long long int mul=0;
    int left,right;
    for(int i=0;i<=n-1;i++)
    {
        int l=0,r=n-1;
        while(l<r)
        {
            int mid=l+r+1>>1;
            if(a[mid]<b[i]) //mid 是合法的，说明区间在右半区间，l = mid
            {
                l=mid; //区间变成[mid, r] 因为mid是合法的。
            }
            else
            {
                r=mid-1;
            }
        }

        if(a[l]>=b[i])
        {
            l=-1;
        }
        left=l;

        l=0,r=n-1;
        while(l<r)
        {
            int mid=l+r>>1;
            if(c[mid]>b[i]) //mid 是合法的，说明区间在左半区间，l = mid
            {
                r=mid;
            }
            else
            {
                l=mid+1;
            }
        }
        if(b[i]>=c[r])
        {
            r=n;
        }
        right=r;
        mul+=1ll*(n-right)*(left+1);
    }
    
    
    /*int a_index = 0, c_index = 0;
    for(int i = 0; i < n; i++)
    {
        int key = b[i];
        while(a_index < n && a[a_index] < key) a_index++;
        while(c_index < n && c[c_index] <= key) c_index++;
        mul += 1ll * (a_index) * (n - c_index);
    }*/
    
    printf("%lld\n",mul);
    return 0;
}