题目描述
小明正在分析一本小说中的人物相关性。他想知道在小说中 Alice 和 Bob 有多少次同时出现。
更准确的说,小明定义 Alice 和 Bob "同时出现" 的意思是:在小说文本 中 Alice 和 Bob 之间不超过 K个字符。
例如以下文本:
This is a story about Alice and Bob.Alice wants to send a private message to Bob.
假设 K = 20,则 Alice 和 Bob 同时出现了 2 次,分别是"Alice and Bob" 和 "Bob. Alice"。前者 Alice 和 Bob 之间有 5 个字符,后者有 2 个字符。
注意:
Alice 和 Bob 是大小写敏感的,alice 或 bob 等并不计算在内。
Alice 和 Bob 应为单独的单词,前后可以有标点符号和空格,但是不能 有字母。例如 Bobbi 並不算出现了 Bob。
输入描述
第一行包含一个整数 K(1 <= K <=10^6)。
第二行包含一行字符串,只包含大小写字母、标点符号和空格。长度不超过 10^6。
输出描述
输出一个整数,表示 Alice 和 Bob 同时出现的次数。
输入输出样例
示例
输入
20 This is a story about Alice and Bob.Alice wants to send a private message to Bob.
输出
2
code
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
const int N = 1000010;
vector<int> Alice,Bob;
int k;
string s;
int main()
{
cin >> k;
cin.get(); // 接受空字符
getline(cin,s);
s += '.'; //最后一个可能不是字符,需加上;如:****Alice
string temp = "";
//找出所有的Alice和Bob并存入Alice和Bob数组;
for (int i = 0 ; i < s.size() ; i ++)
{
if (!isalpha(s[i])) //如果s[i]不是一个字母,那么s[i-1]之前的就是一个单词
{
if(temp == "Alice") Alice.push_back(i - temp.size() );
else if(temp == "Bob") Bob.push_back(i - temp.size() );
temp = "";
}
else temp += s[i];
}
int lp = 0 , rp = 0;
long long ans = 0;
int la = Alice.size();
int lb = Bob.size();
//采用滑动窗口,避免超时,
// 如果采用两重for循环,会超时
for (int i = 0 ; i < la ; i ++)//遍历Alice数组元素
{
while (lp < lb &&Bob[lp] +3+k< Alice[i])//维护窗口左边界,闭区间
lp ++;
while (rp < lb &&Alice[i] + 5 + k >=Bob[rp])//维护窗口右边界,开区间
rp ++;
if (rp - lp > 0) //左闭右开
ans += rp - lp;//答案为窗口中元素个数的累加和
}
cout << ans << endl;
return 0;
}