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 题。

---

解题报告

思路

拼接数字比大小，很自然地想到先排序再双指针找区间端点。（左端点l，右端点r）

• 从左右两端开始进行拼接，先以l为数字的前端。
• 若拼接后得到的数 > k，对r左移，因为在r之前的数全部比r小，拼接后的数也会变小；若对l右移，拼接后的数变大。
• 若拼接后得到的数 < k，对l右移，寻找左边界的最大值。
• 每次找到符合条件的区间[l, r]之后，答案加上r - l，因为在此区间内所有数都比r小，那么与l拼出来的数必定比k小。
• 最后从后往前拼一次

---

AC CODE

#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
#include <algorithm>
#define INF 0x3f3f3f3f
#define LL long long

using namespace std;

int n, m, mina = INF, minb = 2 * INF;
int a[1000006], mark[2005];

int main(){
	cin >> n >> m;
	for(int i = 1; i <= n; ++i){
		scanf("%d", &a[i]);
	}
	mark[a[1]] = 1;
	int cnt = 1;
	for(int l = 1, r = 1; l <= n; ++l){
		mark[a[l - 1]]--;
		if(mark[a[l - 1]] == 0) cnt--;
		while(cnt < m && r <= n){
			r++;
			if(mark[a[r]] == 0){
				cnt++;
			}
			mark[a[r]]++;
		}
		if(cnt == m && r - l + 1 < minb - mina + 1){
			mina = l, minb = r;
		}
	}
	
	cout << mina << ' ' << minb;
	
	return 0;
}