牛客AC题：小葱的01串

链接：登录—专业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);
    }
}