C++递增三元组(枚举、前缀和)

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

本题的关键在于发现每个递增三元组中a[i]和c[k]都要受b[j]约束,故应该通过枚举b[j]计算三元组的个数。同时本题的易错点是统计各个值出现次数的时候,最后一个元素的下标由数据最大值决定,统一用数据范围最大值就好。
AC代码:

#include<stdio.h>
#include<string.h>

const int N=100010;
typedef long long LL;

int n;
int a[N],b[N],c[N];
int cnt[N],s[N];
int anum[N],cnum[N];

int main()
{
    
    
    scanf("%d",&n);
    //因为要计算值出现次数的前缀和,把所有值+1避免占用0下标
    for(int i=1;i<=n;++i){
    
    scanf("%d",&a[i]);++a[i];}
    for(int i=1;i<=n;++i){
    
    scanf("%d",&b[i]);++b[i];}
    for(int i=1;i<=n;++i){
    
    scanf("%d",&c[i]);++c[i];}

    //统计a数组中各个值出现的次数
    for(int i=1;i<=n;++i) ++cnt[a[i]];
    //计算a数组中各个值出现次数的前缀和
    //因为值的范围是1~100001,cnt的最后一项为cnt[100001]
    for(int i=1;i<N;++i) s[i]=s[i-1]+cnt[i];
    //anum[i]中保存a中小于b[i]的数的个数
    for(int i=1;i<=n;++i) anum[i]=s[b[i]-1];

    memset(cnt,0,sizeof(cnt));
    memset(s,0,sizeof(s));
    for(int i=1;i<=n;++i) ++cnt[c[i]];
    for(int i=1;i<N;++i) s[i]=s[i-1]+cnt[i];
    //cnum[i]中保存c中大于b[i]的数的个数
    for(int i=1;i<=n;++i) cnum[i]=s[N-1]-s[b[i]];

    LL ans=0;//统计答案
    for(int i=1;i<=n;++i) ans+=(LL)anum[i]*cnum[i];

    printf("%lld",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_44643644/article/details/108854432