题目:
20
This is a story about Alice and Bob. Alice wants to send a private message to Bob
分析:
字符串长度一百万,肯定不能挨个枚举。
所以,我们可以枚举区间,
首先,我们找出所有的Alice放在Alice集合内
然后,我们找出所有的Bob放在Bob集合内
最后,我们只需要找出每个人名字前面k的区间内有几个另一个人的名字。
区间内人名数量的和就是答案!
为什么只算前面不算后边?
距离是相互的,A到B的距离等于B到A的距离,如果前后计算就重复了。
AC代码:
package JC2019;
import java.util.Scanner;
import java.util.Vector;
public class H {
static boolean check(char C){
if ((C>='a'&&C<='z')||(C>='A'&&C<='Z')) {
return false;
}
return true;
}
public static void main(String[] args) {
Vector<Integer> Alice = new Vector<Integer>();
Vector<Integer> Bob = new Vector<Integer>();
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
String s = sc.nextLine();
s = sc.nextLine();
for (int i = 0; i+5 <= s.length(); i++) {
if ((i==0 || check(s.charAt(i-1))) && (i+5==s.length() || check(s.charAt(i+5)))) {
if (s.substring(i, i+5).equals("Alice")) {
Alice.add(i);
}
}
}
for (int i = 0; i+3 <= s.length(); i++) {
if ((i==0 || check(s.charAt(i-1))) && (i+3==s.length() || check(s.charAt(i+3)))) {
if (s.substring(i, i+3).equals("Bob")) {
Bob.add(i);
}
}
}
int ans = 0;
for (int i = 0; i < Alice.size(); i++) {
int l =0,r=-1;
while(r+1 < Bob.size() && Alice.get(i) > Bob.get(r+1) ) {
r++;
}
while (l<=r && Alice.get(i) >Bob.get(l)+n+3 ) {
l++;
}
ans += r-l+1;
}
for (int i = 0; i < Bob.size(); i++) {
int l =0,r=-1;
while(r+1 < Alice.size() && Bob.get(i) > Alice.get(r+1) ) {
r++;
}
while (l<=r && Bob.get(i) >Alice.get(l)+n+5 ) {
l++;
}
ans += r-l+1;
}
System.out.println(ans);
}
}