前缀和—递增三元组AC

递增三元组描述
在这里插入图片描述前缀和实现思想:

  1. 数据输入:计算的输入数据后自增,便于计算前缀和且不影响其大小关系。
    for(int i = 1;i <= n;i ++)  cin>>a[i],a[i]++;       // 便于下式前缀和的计算
    for(int i = 1;i <= n;i ++)  cin>>b[i],b[i]++;
    for(int i = 1;i <= n;i ++)  cin>>c[i],c[i]++;
  1. 计算AS和CS:根据已知Ai<Bi<Ci,当确定了Bi就能得出A中小于Bi的个数ASi和C中大于Bi的个数CSi,计算CSi*ASi即为遍历Bi时刻得到的三元组个数。
	// 开始计算as[]
    for(int i = 1;i <= n;i ++)  cnt[a[i]] ++;				// 元素出现的个数
    for(int i = 1;i < N;i ++)   s[i] = s[i-1] + cnt[i]; 	// 不大于i的个数
    for(int i = 1;i <= n;i ++)  as[i] = s[b[i]-1];      	// 小于b[i]的数目数组
    // 开始计算cs[]
    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];		// 不大于i的个数
    for(int i = 1;i <= n;i ++)  cs[i] = n - s[b[i]];		// 元素总数-小于或者等于b[i]的数目
  1. 注意点:改题目数据大,所以用long long
res += (long long)as[i]*cs[i];

完整代码:

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

const int N = 1e6;
int as[N],cs[N],a[N],b[N],c[N],s[N],cnt[N];
int n;

int main(){
    cin>>n;
    // 输入数据
    for(int i = 1;i <= n;i ++)  cin>>a[i],a[i]++;       // 便于下式前缀和的计算
    for(int i = 1;i <= n;i ++)  cin>>b[i],b[i]++;
    for(int i = 1;i <= n;i ++)  cin>>c[i],c[i]++;
    // 开始计算as[]
    for(int i = 1;i <= n;i ++)  cnt[a[i]] ++;
    for(int i = 1;i < N;i ++)   s[i] = s[i-1] + cnt[i]; // 不大于i的个数
    for(int i = 1;i <= n;i ++)  as[i] = s[b[i]-1];      // 小于b[i]的数目数组
    // 开始计算cs[]
    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];
    for(int i = 1;i <= n;i ++)  cs[i] = n - s[b[i]];
    // 开始遍历b[]
    long long res = 0;
    for(int i = 1;i <= n;i ++)
        res += (long long)as[i]*cs[i];
    cout<<res<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

滴滴da

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

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

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

打赏作者

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

抵扣说明:

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

余额充值