R（C语言）

demiphea

已于 2022-02-08 18:30:29 修改

阅读量1.7k

点赞数 1

分类专栏： acm个人笔记文章标签：蓝桥杯 c语言

于 2022-02-08 18:28:30 首次发布

本文链接：https://blog.csdn.net/demiphea/article/details/122829086

版权

acm个人笔记专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该博客介绍了如何解决一个字符串处理问题，即在一个只包含大写字母的字符串中找到至少包含k个'R'字符且不包含'P'字符的连续子串的方案数。通过示例和代码解释了如何使用尺取法来解决这个问题，展示了如何在给定的数据范围内高效地计算方案总数。

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

小红拿到了一个长度为 n 的字符串，该字符串仅由大写字母组成。小红很喜欢红色（用'R'字母表示），但她非常讨厌紫色（用'P'字母表示）。她想取一个连续子串，该子串包含至少 k个'R'字符，且不能包含'P'字符。你能告诉她有多少合法的方案可以取到吗？
注：只要连续子串的起始位置或终止位置不同，我们就认为是两个不同的方案。

输入描述:

第一行输入两个正整数 n 和 k ，用空格隔开。
输入一行字符串，该字符串保证仅包含大写字母（'A'到'Z'）。
数据范围：
1≤n≤200000
1≤k≤20

输出描述:

取一个连续子串，包含至少 k 个'R'字符、且不包含'P'字符的方案数。

输入

13 3
RRRPBRRRDBRPR

输出

说明

共有 10 个合法的子串选择方式。假设下标是从 0 到 12 ，那么 10 个合法串分别是：s[0,2]s[0,2]s[0,2] = "RRR"
s[4,7]s[4,7]s[4,7] = "BRRR"
s[4,8]s[4,8]s[4,8] = "BRRRD"
s[4,9]s[4,9]s[4,9] = "BRRRDB"
s[4,10]s[4,10]s[4,10] = "BRRRDBR"
s[5,7]s[5,7]s[5,7] = "RRR"
s[5,8]s[5,8]s[5,8] = "RRRD"
s[5,9]s[5,9]s[5,9] = "RRRDB"
s[5,10]s[5,10]s[5,10] = "RRRDBR"
s[6,10]s[6,10]s[6,10] = "RRDBR"

这些串均含有不小于 3 个字符 'R'，且不包含字符 'P'

输入

5 1
RARPR

输出

单纯靠排列组合，据说用尺取法

#include<stdio.h>
char s[200001];
char s_z[200001];
int x[200001] = { 0 };
int pos[200001];
long long sum = 0;
int main() {
	int n, k;
	scanf("%d %d", &n, &k);
	scanf("%s", s);
	int count = 0, count_r = 0;
	int t = 0;
	for (int i = 0; s[i]; i++) {
		if (s[i] != 'P') {
			count++;
			if (s[i] == 'R') {
				count_r++;
				pos[t++] = count;
			}
			else {
				x[t]++;
			}
		}
		if (s[i] == 'P' || i + 1 == n) {
			if (count_r >= k) {
				for (int j = k - 1; j <= count_r - 1; j++) {
					sum += (x[j - k + 1] + 1) * (count - pos[j] + 1);
				}
			}
			count = 0;
			count_r = 0;
			for (int j = 0; j <= t; j++) {
				x[j] = 0;
			}
			t = 0;
		}
	}
	printf("%lld\n", sum);
	return 0;
}