题目
思路和解题方法
- 程序首先定义了一个函数
check
,用于判断一个字符是否为字母。- 接下来,程序读取输入的整数k和一行字符串str。
- 定义了两个空的向量a和b,用于存储满足条件的子串的起始位置。
- 使用
for
循环遍历字符串str的每个字符,检查是否存在以"Alice"和"Bob"开头的满足条件的子串,并将其起始位置记录在向量a和b中。- 初始化变量ans为0,用于记录满足条件的子串对数。
- 使用双指针技巧遍历向量a中的元素(即以"Alice"开头的子串),维护一个窗口,窗口的左边界为l,右边界为r。
- 在每次遍历时,通过移动右指针r和左指针l的位置,保证窗口内的子串满足要求。具体而言,右指针r向右移动直到超过以"Alice"开头的子串的末尾位置+k+5,左指针l向右移动直到超过以"Alice"开头的子串的起始位置-k-3。
- 每次更新窗口后,将满足条件的子串对数加上窗口中的元素个数(即r-l),并累加到ans中。
- 最后输出ans,即满足条件的子串对数。
复杂度
时间复杂度:
O(n)
- 时间复杂度:假设输入的字符串长度为n,那么代码中的主要操作是遍历字符串、查找子串以及双指针移动。遍历字符串的时间复杂度为O(n),查找子串的时间复杂度为O(n),而双指针移动的时间复杂度为O(n)。因此,总体时间复杂度为O(n)。
空间复杂度
O(n)
- 空间复杂度:代码中使用了两个向量a和b,用于存储满足条件的子串的起始位置。最坏情况下,向量a和b的长度都可能达到n,因此它们所占用的空间复杂度为O(n)。此外,还有一些辅助变量和常数空间的占用,但相对来说是常数级别的。因此,总体空间复杂度为O(n)。
c++ 代码
cpp
#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;
bool check(char c){
return c>='A'&&c<='Z'||c>='a'&&c<='z';
}
int main(){
ll k;
cin>>k;
getchar();
string str;
getline(cin,str);
vector<ll> a,b;
// 遍历字符串,找出Alice和Bob的位置
for(ll i = 0;i+3<str.size();i++)
{
// 判断是否为Alice
if(str.substr(i,5) == "Alice" && (!check(str[i+5]))&&(!check(str[i-1])) && i+5<str.size())
a.push_back(i);
// 判断是否为Bob
if(str.substr(i,3) == "Bob" && (!check(str[i+3]))&&(!check(str[i-1])))
b.push_back(i);
}
ll ans =0 ;
for(ll i = 0,l=0,r=0; i < a.size(); i++){ //遍历Alice数组元素
while(l < b.size() && b[l] < a[i] - k - 3) l++; //维护窗口左边界
while(r < b.size() && b[r] <= a[i] + k + 5) r++; //维护窗口右边界
ans += r - l; //答案加上窗口中元素个数
}
cout<<ans<<endl;
}
cpp
注释:
ll 是 long long 类型的别名,用于存储较大的整数值。
check() 函数用于判断一个字符是否为字母。
k 是输入的一个整数值,表示一个窗口的大小。
str 是输入的字符串。
a 和 b 分别是存储 Alice 和 Bob 出现位置的数组。
遍历字符串 str,找出每个 Alice 和 Bob 的位置,并将其存储在对应的数组中。
ans 是计数变量,记录符合条件的情况个数。
使用两个指针 l 和 r 维护一个滑动窗口。
遍历数组 a,对于每个 Alice 的位置,更新滑动窗口的左右边界。
l 和 r 分别指向滑动窗口的左右边界。
ans 加上窗口中元素的个数,即 r - l。
最后输出答案 ans。
觉得有用的话可以点点赞,支持一下。
如果愿意的话关注一下。会对你有更多的帮助。
每天都会不定时更新哦 >人< 。