递增三元组

1236. 递增三元组 - AcWing题库

给定三个整数数组

A=[A1,A2,…AN]A=[A1,A2,…AN],
B=[B1,B2,…BN]B=[B1,B2,…BN],
C=[C1,C2,…CN]C=[C1,C2,…CN],

请你统计有多少个三元组 (i,j,k)(i,j,k) 满足：

1. 1≤i,j,k≤N1≤i,j,k≤N
2. Ai<Bj<CkAi<Bj<Ck

输入格式

第一行包含一个整数 NN。

第二行包含 NN 个整数 A1,A2,…ANA1,A2,…AN。

第三行包含 NN 个整数 B1,B2,…BNB1,B2,…BN。

第四行包含 NN 个整数 C1,C2,…CNC1,C2,…CN。

输出格式

一个整数表示答案。

数据范围

1≤N≤1051≤N≤105,
0≤Ai,Bi,Ci≤1050≤Ai,Bi,Ci≤105

输入样例：

3
1 1 1
2 2 2
3 3 3

输出样例：

27

思路：

数据范围十万，n或nlogn，只能枚举ABC中的一组。枚举AC后都会剩下两组，互相约束，因此枚举B。

做法1：排序+二分

排序A、C，枚举B， 在A二分找小于Bi的元素个数，在C中找大于Bi的元素个数，加和。

#include<iostream>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N = 1e5+10;
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);
    
    //枚举
    LL ans = 0;
    for(int i=0;i<n;i++){
        int l = 0, r = n - 1;
        while(l<r){
            int mid = l+r+1>>1;
            if(a[mid]<b[i]) l = mid;
            else r = mid - 1;
        }
        if(a[l]>=b[i]) l = -1;
        int x = l;
        
        l = 0, r = n-1;
        while(l<r){
            int mid = l+r>>1;
            if(c[mid]>b[i]) r = mid;
            else l = mid + 1;
        }
        if(c[l]<=b[i]) l = n;
        int y = l;
        ans += (LL)(x+1)*(n-y);
    }
    cout<<ans<<endl;
    return 0;
}

做法2：前缀和

cnt [ i ]表示在A中，i这个值出现多少次。

for(int i=0;i<n;i++)
    cnt[A[i]]++;

S[ i ] = cnt[ 0 ] + cnt[ 1 ] + ... + cnt[ i ]

表示在A中，0~i出现多少次

另外，前缀和、dp等需要用到下标-1操作的算法，在存数据时下标++，避免额外处理一些边界情况。

#include<iostream>
#include<cstring>
using namespace std;

typedef long long LL;

const int N = 1e5+10;
int a[N],b[N],c[N];
int cnt[N],s[N];
int as[N],cs[N];

int main(){
    int n;
    cin>>n;
    for(int i=0;i<n;i++) {cin>>a[i];a[i]++;};
    for(int i=0;i<n;i++) {cin>>b[i];b[i]++;};
    for(int i=0;i<n;i++) {cin>>c[i];c[i]++;};
    
    //求A中大于Bi的数量
    for(int i=0;i<n;i++) cnt[a[i]]++;
    for(int i=1;i<N;i++) s[i] = s[i-1] + cnt[i];
    for(int i=0;i<n;i++) as[i] = s[b[i] - 1];//前面读入时++，防止这里前缀和时空指针异常
    
    memset(cnt,0,sizeof cnt);
    memset(s,0,sizeof s);
    
    //求C中大于Bi的数量
    for(int i=0;i<n;i++) cnt[c[i]]++;
    for(int i=1;i<N;i++) s[i] = s[i-1] + cnt[i];
    for(int i=0;i<n;i++) cs[i] = s[N - 1] - s[b[i]];
    
    //枚举
    LL ans = 0;
    for(int i=0;i<n;i++){
        ans += (LL)as[i]*cs[i];
    }
    cout<<ans<<endl;
    return 0;
}