蓝桥杯练习—递增三元组

天亮之前_ict

已于 2023-01-18 14:59:22 修改

阅读量281

点赞数

文章标签：蓝桥杯算法职场和发展

于 2023-01-11 13:18:01 首次发布

原文链接：https://www.codeheng.top/?post=48

版权

文章介绍了如何使用C++编程解决一个数组问题，涉及到对三个已排序数组A、B、C进行处理，找到A中小于B[i]的元素数量和C中大于B[i]的元素数量，然后计算总和。文章提到了使用标准库的sort函数进行排序，并利用lower_bound和upper_bound进行二分查找，以及自定义二分查找函数的方法。

摘要由CSDN通过智能技术生成

题目描述

在这里插入图片描述

输入描述

在这里插入图片描述

输出描述

输出一个整数表示答案。

输入输出样例

示例

输入

输出

分析

首先对题目进行分析，三个数组A、B、C里元素的排列顺序是A < B < C，所以说应该以B为中间数进行寻找，为了降低时间复杂度，应使用二分查找的方式寻找A中小于B元素的个数与C中大于B元素的个数，在查找前，需要对个数组进行排序，在这里用到了C++排序函数`sort()`进行排序，即：

sort(a, a + n);    
sort(b, b + n);    
sort(c, c + n);

随后进行二分查找，这里有两种方式

使用C++函数lower_bound()、upper_bound()、binary_search()进行查找
- lower_bound()：返回大于或等于目标值的第一个位置
- per_bound()：返回大于目标值的第一个位置
- nary_search()：若目标值存在则返回true，否则返回false

相关代码如下：

long long fina l= 0;
    for (int i = 0; i < n; i++) { 
        ll x = lower_bound(a, a + n, b[i]) - a;
		// lower_bound二分查找第一个大于或等于b[i]的数字

        ll y = n - (upper_bound(c, c + n, b[i]) - c);
		// upper_bound二分查找第一个大于b[i]的数字

        final += y*x;//统计符合条件的个数
    }

使用自定义函数进行二分查找，代码如下：

int find1(int a[],int goal,int end){
    int left=0;
    int right=end;
    while(left<=right){
        int mid=(left+right)/2;
        if(goal>a[mid]) {
            left=mid+1;
        }else{
            right=mid-1;
        }
    }
    return left;
}
int find2(int a[],int goal,int end){
    int left=0;
    int right=end;
    while(left<=right){
        int mid=(left+right)/2;
        if(goal>=a[mid]) {
            left=mid+1;
        }else{
            right=mid-1;
        }
    }
    return left;
}

完整代码

第一种

#include <iostream>
#include <algorithm>
#define ll long long

using namespace std;

int main() { 
    int n;    
	cin >> n;
	int a[n];
	int b[n];
	int c[n];
    for (int i = 0; i < n; i++){
	scanf("%d", &a[i]);
    }
    for (int i = 0; i < n; i++){ 
	scanf("%d", &b[i]);
	}
    for (int i = 0; i < n; i++){ 
	scanf("%d", &c[i]);
	}
    sort(a, a + n);    
	sort(b, b + n);    
	sort(c, c + n);
    ll final = 0;
    for (int i = 0; i < n; i++) { 
        ll x = lower_bound(a, a + n, b[i]) - a;// lower_bound二分查找第一个大于或等于b[i]的数字

        ll y = n - (upper_bound(c, c + n, b[i]) - c);// upper_bound二分查找第一个大于b[i]的数字

        final += y*x;
    }
    cout << final;
}

第二种

#include<iostream>
#include<algorithm>
using namespace std;
int find1(int a[],int goal,int end){
    int left=0;
    int right=end;
    while(left<=right){
        int mid=(left+right)/2;
        if(goal>a[mid]) {
            left=mid+1;
        }else{
            right=mid-1;
        }
    }
    return left;
}
int find2(int a[],int goal,int end){
    int left=0;
    int right=end;
    while(left<=right){
        int mid=(left+right)/2;
        if(goal>=a[mid]) {
            left=mid+1;
        }else{
            right=mid-1;
        }
    }
    return left;
}
int main(){
    int n;
    cin>>n;
    int A[n],B[n],C[n];
    long long ans=0;
    for(int i=0;i<n;i++){
        cin>>A[i];
    }
    for(int i=0;i<n;i++){
        cin>>B[i];
    }    
    for(int i=0;i<n;i++){
        cin>>C[i];
    }
    sort(A,A+n);
    sort(B,B+n);
    sort(C,C+n);
    for(int i=0;i<n;i++){
        long long x=find1(A,B[i],n-1);
        long long y=n-find2(C,B[i],n-1);
        ans+=y*x;
    }
    cout<<ans;
    return 0；
}