题目
给一个带有 B*A*C*D*
模式的字符串,*
意味着前面的字符在字符串中可以显示 0次
或 多次
。计算字符 'A' 出现的次数。
说明
输入: s = "BBAACCDDD"
输出: 2
输入: s = "BBCDD"
输出: 0
字符顺序为:B A C D
答案
private static int bsStart(String s) {
int start = 0;
int end = s.length() - 1;
int mid = start + (end - start) / 2;
char tmp = 'A';
while (start + 1 < end) {
tmp = s.charAt(mid);
if (tmp == 'B') {
start = mid + 1;
} else if (tmp == 'A' || tmp > 'B') {
end = mid;
}
mid = start + (end - start) / 2;
}
if (s.charAt(start) == 'A') {
return start;
}
return end;
}
private static int bsEnd(String s) {
int start = 0;
int end = s.length() - 1;
int mid = start + (end - start) / 2;
char tmp = 'A';
while (start + 1 < end) {
tmp = s.charAt(mid);
if (tmp <= 'B') {
start = mid;
} else if (tmp > 'B') {
end = mid - 1;
}
mid = start + (end - start) / 2;
}
if (s.charAt(end) == 'A') {
return end;
}
return start;
}
public static int countA(String s) {
// Write your code here
int length = s.length();
if (length == 0) {
return 0;
}
int start = 0, end = length - 1;
if (s.charAt(0) != 'A') {
start = bsStart(s);
}
if (s.charAt(start) != 'A') {
return 0;
}
if (s.charAt(end) != 'A') {
end = bsEnd(s);
}
return end - start + 1;
}
public static void main(String[] args) {
System.out.println(Test.countA("BBBBBAAAAAAAACDDDDD"));
}
解析
使用二分法来找出字符 A 出现的 起始位置 和 结束位置计算 A 出现的次数