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;
}
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

赴星辰大海

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值