蓝桥杯 递增三元组 C语言

题目描述

给定三个整数数组

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 个整数 1,2,⋯A1,A2,⋯AN。

第三行包含 N 个整数 1,2,⋯B1,B2,⋯BN。

第四行包含 N 个整数 1,2,⋯C1,C2,⋯CN。

其中，1≤≤105,0≤,,≤1051≤N≤105,0≤Ai,Bi,Ci≤105。

输出描述

输出一个整数表示答案。

输入输出样例

示例

输入

3
1 1 1
2 2 2
3 3 3

输出

27

运行限制

• 最大运行时间：2s

• 最大运行内存: 256M

思路

A、B、C有序方便计算个数而非一个个查找看有几个，对每个B 找到符合要求的A数量 和 C数量 一遍遍历结束

#include <stdio.h>
#include <stdlib.h>

int a[100005], b[100005], c[100005], n;

int cmp(const void* p, const void* q);
int BinSearch_low(int key);
int BinSearch_high(int key);

int main(int argc, char* argv[]) {
    int i, hig, low;
    long long sum = 0;
    scanf("%d", &n);
    for (i = 0; i < n; i++) scanf("%d", &a[i]);
    qsort(a, n, sizeof(a[0]), cmp);
    for (i = 0; i < n; i++) scanf("%d", &b[i]);
    qsort(b, n, sizeof(b[0]), cmp);
    for (i = 0; i < n; i++) scanf("%d", &c[i]);
    qsort(c, n, sizeof(c[0]), cmp);

    for (i = 0; i < n; i++) {
        if (b[i] <= a[0]) continue;
        if (b[i] >= c[n-1]) break;
        low = BinSearch_high(b[i] - 1);
        hig = BinSearch_low(b[i] + 1);
        sum += (long long)(low + 1) * (long long)(n - hig);
    }
    printf("%lld\n", sum);
    return 0;
}

int cmp(const void* p, const void* q) {
    int* a = (int*)p;
    int* b = (int*)q;
    if(*a > *b) return 1;
    else return -1;
}

int BinSearch_low(int key) {
    int low = 0, high = n - 1, mid;
    while (high >= low) {
        mid = (high + low) / 2;
        if (c[mid] >= key) high = mid - 1;
        else low = mid + 1;
    }
    return low;
}

int BinSearch_high(int key) {
    int low = 0, high = n - 1, mid;
    while (high >= low) {
        mid = (high + low) / 2;
        if (a[mid] <= key) low = mid + 1;
        else high = mid - 1;
    }
    return high;
}