小明正在分析一本小说中的人物相关性。他想知道在小说中 Alice 和 Bob 有多少次同时出现。
更准确的说,小明定义 Alice 和 Bob “同时出现” 的意思是:在小说文本 中 Alice 和 Bob 之间不超过 KK 个字符。
例如以下文本:
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 <= 1e6)。
第二行包含一行字符串,只包含大小写字母、标点符号和空格。长度不超过 1e6。
输出描述
输出一个整数,表示 Alice 和 Bob 同时出现的次数。
示例
输入:
20
This is a story about Alice and Bob.Alice wants to send a private message to Bob.
输出:
2
思路:
- 对于每一个出现的Alice或Bob,用一个数组nums存储,nums[0]表示起始字符,nums[1]表示结束字符。所有的Alice按起始字符从小到大存储在列表list1中,Bob按起始字符从小到大存储到列表list2中。
- 第一想法是遍历每一个Alice,判断[nums[0] - k, nums[0] - 1]中和[nums[1] + 1, nums[1] + k]中有多少个Bob再加到最终的答案中。这是一个很朴素的想法,同时也是可以通过80%的样例的。但相信读者应该和我一样有这样的想法,假如Alice1号出现在Alice2号之间,那么Alice1的左边访问不到的Bob,Alice2肯定也访问不到,同时Alice2还可以访问到比Alice1更右边的Bob,但不知道该如何用代码表达。是定义临时变量,还是?经过看其他博客的提示,咋一看,随着list1的遍历,能访问的到Bob的左边界往左移,右边界往右移,还满足一定的限制,这是不是正符合滑动窗口的定义吗,附AC代码。
import java.util.*;
public class 人物相关性 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int k = in.nextInt();
in.nextLine();
String s = in.nextLine();
int idx = 0;
List<int[]> list1 = new ArrayList<int[]>(), list2 = new ArrayList<int[]>();
while (idx != -1) {
idx = s.indexOf("Alice", idx);
if (idx == -1) {
break;
}
if (idx - 1 >= 0 && Character.isLetter(s.charAt(idx - 1)) || idx + 4 < s.length() && Character.isLetter(s.charAt(idx +5))) {
idx += 5;
continue;
}
list1.add(new int[] {idx, idx + 4});
idx += 5;
}
idx = 0;
while (idx != -1) {
idx = s.indexOf("Bob", idx);
if (idx == -1) {
break;
}
if (idx - 1 >= 0 && Character.isLetter(s.charAt(idx - 1)) || idx + 3 < s.length() && Character.isLetter(s.charAt(idx +3))) {
idx += 3;
continue;
}
list2.add(new int[] {idx, idx + 2});
idx += 3;
}
long ans = 0;
int l = 0, r = 0;
for (int i = 0; i < list1.size(); i++) {
while (l < list2.size() && list2.get(l)[1] < list1.get(i)[0] - k) {
l++;
}
while (r < list2.size() && list2.get(r)[0] <= list1.get(i)[1] + k) {
r++;
}
ans += r - l;
}
System.out.println(ans);
}
}
|