2018蓝桥杯B组第6题：递增三元组

题目描述：递增三元组

给定三个整数数组
A = [A1, A2, ... AN],
B = [B1, B2, ... BN],
C = [C1, C2, ... CN]，
请你统计有多少个三元组(i, j, k) 满足：

1. 1 <= i, j, k <= N
2. Ai < Bj < Ck

【输入格式】
第一行包含一个整数N。
第二行包含N个整数A1, A2, ... AN。
第三行包含N个整数B1, B2, ... BN。
第四行包含N个整数C1, C2, ... CN。

对于30%的数据，1 <= N <= 100
对于60%的数据，1 <= N <= 1000
对于100%的数据，1 <= N <= 100000 0 <= Ai, Bi, Ci <= 100000

【输出格式】
一个整数表示答案

【样例输入】
3
1 1 1
2 2 2
3 3 3

【样例输出】
27

---

思路

相对比较简单的一道题

看数据范围就知道O(N^2)的算法是行不通的，就只能O(N*logN)了。

先对三个组都进行排序（用STL中的sort快排）

循环B中的每一个数，保证不重复

lower_bound 二分查找A中第一个大于等于B[i]的数

upper_bound 二分查找C中第一个大于B[i]的数

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>

using namespace std;
typedef long long ll;
const int maxn = 100000+10;
vector<int> info[maxn];
int ans[maxn];
int n,d,k;
bool check(int cnt){
    int len = info[cnt].size();
    if(len<k)return false;
    int l=0,r=0,sum=0;
    sort(info[cnt].begin(),info[cnt].end());
    while(l<=r&&r<len){
        sum++;
        if(sum>=k){
            if(info[cnt][r]-info[cnt][l]<d){
                return true;
            }   
            else{
                sum--;
                l++;
            }
        }
        r++;
    }
    return false;
}
int main(){
    int ts,id;
    scanf("%d%d%d",&n,&d,&k);
    for(int i = 0;i<n;i++){
        scanf("%d%d",&ts,&id);
        info[id].push_back(ts);
    }
    int all = 0;
    for(int i = 0;i<=100000;i++){
        if(check(i)){
            ans[all++]=i;
        }
    }
    for(int i = 0;i<all;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}