蓝桥杯——人物性格相关分析(滑动窗口)
题目描述
小明正在分析一本小说中的人物相关性。他想知道在小说中 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 同时出现的次数。
输入输出样例
示例
输入
20This is a story about Alice and Bob.Alice wants to send a privatemessage to Bob.
输出
2
code
#include #include #include #include using namespace std;const int N = 1000010;vector 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 =Bob[rp])//维护窗口右边界,开区间 rp ++; if (rp - lp > 0) //左闭右开 ans += rp - lp;//答案为窗口中元素个数的累加和 } cout << ans << endl; return 0; }