人物相关性JavaB组(滑动窗口解决)

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

注意:

  1. Alice 和 Bob 是大小写敏感的,alice 或 bob 等并不计算在内。
  2. 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;
			}
			// System.out.println(idx);
			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))) {
				// System.out.println(true + " " + s.charAt(idx - 1) + " " + 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);
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ZW钟文

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值