不知道哪里的题:Three Arrays(二分+计数)

题目大意:

        给你三个数组a,b,c(非降序排列),分别包含na, nb, nc个整数。问你能找到多少个这样的三元组(i, j, k),使得ai, bj, ck三者两两之间绝对值之差不超过d。

解题思路:

       显然ai, bj, ck这三个数必然在区间[i, i+d]内。有三种计数情况:1:ai <= bj <= ck ; 2 : ai > bj 且 ck >= bj;3: ai > ck 且 bj > ck。(可以理解为排列的开头不同)根据以上三种情况进行计数。要迅速找出一个区间内有多少个数,二分即可。只需要注意lower_bound和upper_bound分别就好 。

代码如下:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[500005], b[500005], c[500005];
int n, A, B, C;
int main(){
    while(scanf("%d %d %d %d", &n, &A, &B, &C) != EOF){
        for(int i=0;i<A;++i) scanf("%d", &a[i]);
        for(int i=0;i<B;++i) scanf("%d", &b[i]);
        for(int i=0;i<C;++i) scanf("%d", &c[i]);
        ll ans = 0;
        for(int i=0;i<A;++i){
            ll cnt1 = upper_bound(b, b+B, a[i]+n) - lower_bound(b, b+B, a[i]);
            ll cnt2 = upper_bound(c, c+C, a[i]+n) - lower_bound(c, c+C, a[i]);
            ans += cnt1*cnt2;
    //        cout << cnt1*cnt2 << endl;
        }
        for(int i=0;i<B;++i){
            ll cnt1 = upper_bound(a, a+A, b[i]+n) - upper_bound(a, a+A, b[i]);
            ll cnt2 = upper_bound(c, c+C, b[i]+n) - lower_bound(c, c+C, b[i]);
            ans += cnt1*cnt2;
   //         cout << cnt1*cnt2 << endl;
        }
        for(int i=0;i<C;++i){
            ll cnt1 = upper_bound(b, b+B, c[i]+n) - upper_bound(b, b+B, c[i]);
            ll cnt2 = upper_bound(a, a+A, c[i]+n) - upper_bound(a, a+A, c[i]);
            ans += cnt1*cnt2;
     //       cout << cnt1*cnt2 << endl;
        }
        cout << ans << endl;
    }
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值