Python 人物相关性分析-字符串问题（来源 lanqiao.cn:198）

题目描述
小明正在分析一本小说中的人物相关性。他想知道在小说中 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 个字符。 注意:

  1.Alice 和 Bob 是大小写敏感的，alice 或 bob 等并不计算在内。
  2.Alice 和 Bob 应为单独的单词，前后可以有标点符号和空格，但是不能

有字母。例如 Bobbi 並不算出现了 Bob。

输入格式
第一行包含一个整数 K。 第二行包含一行字符串，只包含大小写字母、标点符号和空格。长度不超 过 1000000。 (对于所有评测用例，1≤ K ≤1000000。)
输出格式
输出一个整数，表示 Alice 和 Bob 同时出现的次数
样例输入
20 This is a story about Alice and Bob. Alice wants to send a private message to Bob.
样例输出
2

知识补充：
Python自带库bisect，列表必须有序
bisect.bisect_left和bisect.bisect_right
假设arr = [1, 2, 3, 5]
bisect.bisect_left(arr, 0) = 0
bisect.bisect_left(arr, 1) = 0
bisect.bisect_right(arr, 0) = 0
bisect.bisect_right(arr, 1) = 1
当查找到对应元素，返回该元素位置，查不到的话，left返回讲该元素插入时候的位置

解题思路：
开始的思路有点问题，是直接在for循环中找到对应的字符后就开始判断，但是这样符合的结果太多了。后面看了其他人的一个思路，然后加入了python自带的库 bisect。
首先找到对应单词所在的位置，满足条件保存该索引值，之后遍历alice所在的列表向第二个bob列表查询。
a1 代表当前元素减去最远距离 + 3 (len(“Bob”))
a2 代表当前元素加上最远距离 + 5 (len(“Alice”))
如果a1 < 0，那么查找后肯定是0，只需要查找a2所在的索引
a1 > 0, 查找a2 和 a1 在bob中的索引并相减，如果两个值对应的索引符合的话，对应相加，否则则会认定找不到。例如：a = 36时，a1 = 36 - 23 = 13, a2 = 36 + 25 = 61, 最后得到的结果是1 - 0 = 1只有一个符合，也就是bob中的 32满足条件

代码：

import bisect
k = int(input())
s = input()

x1 = "Alice"
x2 = "Bob"
pun = [',', '.', '!', '?', ' '] #标点
alice = []
bob = []
cnt = 0 #总数

for i in range(len(s)):
	#判断接下来的5 or 3个字符是否匹配，以及前面一个字符是否是标点
    if s[i] == 'A':
        if s[i: i + 5] == x1 and (s[i - 1] in pun):
            alice.append(i)
    if s[i] == 'B':
        if s[i: i + 3] == x2 and (s[i - 1] in pun):
            bob.append(i)
# alice = [22, 36]; bob = [32, 77]
for a in alice:
    a1 = a - (k + 3)
    a2 = a + k + 5
    # 如果a1小于0，那么只需要找a2在bob中的索引，因为a1查找出来肯定是0
    if a1 < 0:
        cnt += bisect.bisect_left(bob, a2)
    else:
        cnt += bisect.bisect_left(bob, a2) - bisect.bisect_left(bob, a1)
print(cnt)