链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
小葱的01串
题目描述
给定一个长度为偶数的环形 01 字符串。(环形指,第一个字符和最后一个字符是相邻的)
字符串初始每个字符都是白色。小葱想把一段连续区间染成红色,使得红色的字符'0'数量等于白色的字符'0'数量,红色的字符'1'数量等于白色的字符'1'数量。问有多少种不同的染色方法?
两个方案不同当且仅当存在一个某字符,在一个方案是染成红色,在另一个方案为白色。
算法原理
从题目可以分析出来,需要一段连续区间来进行染色,并且出现了字符数量,所以我们就可以通过滑动窗口 + 哈希表来解决问题,由于题目说明是一个环形字符串,所以要进行特殊处理,当right等于数组长度时,让它回到0下标就可以了
题目代码
import java.util.Scanner;
public class test37 {
// 算法原理:滑动窗口 + 哈希表
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
// 处理输入
int n = in.nextInt();
String s = in.next();
// 模拟哈希表
int[] hash1 = new int[2];
int[] hash2 = new int[2];
for (int i = 0; i < n; i++) {
hash1[s.charAt(i) - '0']++;
}
int ret = 0;
// 进窗口
for (int left = 0, right = 0; left < n; ) {
char ch = s.charAt(right);
hash2[ch - '0']++;
// 判断 + 更新结果
if (hash2[0] * 2 == hash1[0] && hash2[1] * 2 == hash1[1]) {
ret++;
}
// 出窗口
while ((hash2[0] + hash2[1]) * 2 == s.length()) {
char out = s.charAt(left++);
hash2[out - '0']--;
}
right++;
// 处理环形的情况
if (right == n) right = 0;
}
System.out.println(ret);
}
}