整数小拼接--二分法

PL_涵

已于 2023-03-14 19:09:56 修改

阅读量834

点赞数

分类专栏：【代码实战例题】文章标签：蓝桥杯

于 2022-11-21 17:51:20 首次发布

本文链接：https://blog.csdn.net/PL_hfc/article/details/127969169

版权

【代码实战例题】专栏收录该内容

6 篇文章 0 订阅

订阅专栏

原题链接：蓝桥杯2020年第十一届省赛真题-整数小拼接 - C语言网 (dotcpp.com)

题目描述

给定一个长度为 $n$ 的数组 $A_{1},A_{2},\cdots ,A_{n}$ 。你可以从中选出两个数 $A_{i},A_{j} (i\neq j)$ ，然后将 $A_{i}$ 和 $A_{j}$ 一前一后拼成一个新的整数。例如 12 和 345 可以拼成 12345 或 34512 。注意交换 $A_{i}$ 和 $A_{j}$ 的顺序被视为 2 种拼法，即便是 $A_{i}=A_{j}$ 。

请你计算有多少种拼法满足拼出的整数小于等于 $K$ 。

输入格式

第一行包含 2 个整数 $n$ 和 $K$ 。第二行包含 $n$ 个整数 $A_{1},A_{2},\cdots ,A_{n}$ 。

输出格式

一个整数代表答案。

样例输入

4 33
1 2 3 4

样例输出

第一种解法：暴力枚举 30pts

使用两层循环迭代所有可能的组合，并判断连在一起是否小于 K 。

可以估计出算法的时间复杂度为 $O(n^{2})$ ，也就是只能通过30%的用例，得分 30 pts 。

#include <bits/stdc++.h>
using namespace std;
const int MAX_N = 100100;
int n;
long long k, a[MAX_N], ans;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
    }
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            if (i == j) {
                continue;
            }
            if (a[i] * (long long)pow(10, int(log10(a[j]) + 1)) + a[j] <= k) {
                ans ++;
            }
        }
    }
    cout << ans << endl;
    return 0;
}

第二种解法：二分法 100pts

即首先将原数组先排序，然后用一层循环枚举每一个 $A_{i}$ ，然后利用二分查找，找出能够满足条件的 $A_{j}$ ，估计时间复杂度为 $O(nlogn)$ 。

拼接两个整数时，可以使用 $log_{10}(x)+1$ 来计算整数 $X$ 的位数。

二分查找可以自己写一个，也可以选择直接用C++函数库里的 。

注意：不要忘记开 $longlong$ ，因为拼接后的数可以很大。

#include <bits/stdc++.h>
using namespace std;
const int MAX_N = 100100;
int n;
long long k, a[MAX_N], ans;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
    }
    sort(a + 1, a + n + 1);
    for (int i = 1; i <= n; ++i) {
        if (a[i] > k % (long long)pow(10, int(log10(a[i]) + 1))) {
            ans += upper_bound(a + i + 1, a + n + 1, k / (long long)pow(10, int(log10(a[i]) + 1)) - 1) - 1 - a - i;
            ans += upper_bound(a + 1, a + i, k / (long long)pow(10, int(log10(a[i]) + 1)) - 1) - 1 - a;
        } else {
            ans += upper_bound(a + i + 1, a + n + 1, k / (long long)pow(10, int(log10(a[i]) + 1))) - 1 - a - i;
            ans += upper_bound(a + 1, a + i, k / (long long)pow(10, int(log10(a[i]) + 1))) - 1 - a;
        }
    }
    cout << ans << endl;
    return 0;
}