题目来源:第十四届蓝桥杯大赛软件赛省赛B组

程序猿圈子里正流行一种很新的简写方法:对于一个字符串,只保留首位字符,将首位字符之间的所有字符用这部分的长度代替。例如internationalization简写成i18n,Kubernetes简写成K8s,Lanqiao简写成L5o等。

在本题中,我们规定长度大于等于 KKK 的字符串都可以采用这种简写方法(长度小于 KKK 的字符串不配使用这种缩写)

给定一个字符串 SSS 和两个字符 c1c_1c1c2c_2c2 , 请你计算 SSS 有多少个以 c1c_1c1 开头 c2c_2c2 结尾的子串可以采用这种缩写?

输入: 第一行包含一个整数 KKK

第二行包含一个字符串 SSS 和两个字符 c1c_1c1c2c_2c2

输出: 一个整数代表答案

Input Sample:

4
abababdb a b

Output Sample:

6

先说这道题的解题方法: 二分法or双指针, 两种方案的题解我都会给出来

首先先把字符串中 c1c_1c1c2c_2c2 的下标单独用两个vector进行维护, 我们假设维护 c1c_1c1 下标的vector为 aaa , 维护 c2c_2c2 的为 bbb , 那么我们要遍历 aaa , 假设此时指向 aaa 的指针为 iii , 指向 bbb 的指针为 jjj , 我们要在这个循环内部用双指针或者二分法来查找到使得 j−i+1=Kj - i+ 1 = Kji+1=K 的这个 jjj

下面给出题解代码, 共两个方案, 但核心思路都是一样的. 请注重思考, 不要无脑cv

方案1: 双指针法

#include <bits/stdc++.h>
using namespace std;
const int maxn = 55;
vector<int> a, b;

void io() {
	cin.tie(0);
	cout.tie(0);
	ios::sync_with_stdio(false);
}

int main() {
	io();
	int k;
	cin >> k;
	string s;
	char c1, c2;
	cin >> s >> c1 >> c2;
	int res = 0;
	for (int i = 0; i < s.length(); i++) {
		if (s[i] == c1) {
			a.push_back(i);
		}
		if (s[i] == c2) {
			b.push_back(i);
		}
	}
	for (int i = 0, j = 0; i < a.size(); i++) {
		while (j < b.size() && b[j] - a[i] + 1 < k) {
			j++;
		}
		res += (b.size() - j);
	}
	cout << res << '\n';
	return 0;
}

方案2: 二分法

#include <bits/stdc++.h>
using namespace std;
const int maxn = 55;
vector<int> a, b;

void io() {
	cin.tie(0);
	cout.tie(0);
	ios::sync_with_stdio(false);
}

int main() {
	io();
	int k;
	cin >> k;
	string s;
	char c1, c2;
	cin >> s >> c1 >> c2;
	int res = 0;
	for (int i = 0; i < s.length(); i++) {
		if (s[i] == c1) {
			a.push_back(i);
		}
		if (s[i] == c2) {
			b.push_back(i);
		}
	}
	for (int i = 0; i < a.size(); i++) {
		while (l <= r) {
			int mid = l + r >> 1;
			if (b[mid] - a[i] + 1 > k) {
				r = mid - 1;
			} else if (b[mid] - a[i] + 1 == k) {
				res += (b.size() - mid);
				break;
			} else {
				l = mid + 1;
			}
		}
	}
	cout << res << '\n';
	return 0;
}
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐