P8708 [蓝桥杯 2020 省 A1] 整数小拼接（二分）

[蓝桥杯 2020 省 A1] 整数小拼接

题目描述

给定一个长度为 $n$ 的数组 $A_1,A_2,\cdots ,A_n$。你可以从中选出两个数 $A_i$ 和 $A_j$($i\ne 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$。

输出格式

一个整数代表答案。

样例 #1

样例输入 #1

4 33
1 2 3 4

样例输出 #1

8

提示

对于 $30\%$ 的评测用例 $1\le n\le 1000$，$1\le k\le 10^8$，$1\le A_i\le 10^4$。

对于所有评测用例，$1\le n\le 10^5$，$1\le k\le 10^{10}$，$1\le A_i\le 10^9$。

蓝桥杯 2020 第一轮省赛 A 组 H 题。

解析

二分，对于左边部分序列肯定可以拼接，右边不可以拼接，二分找分割点

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define inf 1e18
const int mod=1e9+7;
const int N=2e5+5;
int n,k,a[N];
int f(int a,int b){
	return stoll(to_string(a)+to_string(b));
}
void solve(){
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	sort(a+1,a+n+1);
	int ans=0;
	for(int i=1;i<=n;i++){
		int l=0,r=n,t=0;
		while(l<=r){
			int mid=l+r>>1;
			if(f(a[i],a[mid])<=k) t=mid,l=mid+1;
			else r=mid-1;
		}
		if(t>=i) ans+=t-1;
		else ans+=t;
	}
	cout<<ans;
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	int tt=1;
//	cin>>tt;
	while(tt--) solve();
	return 0;
}