[LeetCode]2222. 选择建筑的方案数

题目

2222. 选择建筑的方案数
给你一个下标从 0 开始的二进制字符串 s ，它表示一条街沿途的建筑类型，其中：

s[i] = '0' 表示第 i 栋建筑是一栋办公楼，
s[i] = '1' 表示第 i 栋建筑是一间餐厅。
作为市政厅的官员，你需要随机 选择 3 栋建筑。然而，为了确保多样性，选出来的 3 栋建筑 相邻 的两栋不能是同一类型。

比方说，给你 s = "001101" ，我们不能选择第 1 ，3 和 5 栋建筑，因为得到的子序列是 "011" ，有相邻两栋建筑是同一类型，所以 不合 题意。
请你返回可以选择 3 栋建筑的 有效方案数 。

 

示例 1：

输入：s = "001101"
输出：6
解释：
以下下标集合是合法的：
- [0,2,4] ，从 "001101" 得到 "010"
- [0,3,4] ，从 "001101" 得到 "010"
- [1,2,4] ，从 "001101" 得到 "010"
- [1,3,4] ，从 "001101" 得到 "010"
- [2,4,5] ，从 "001101" 得到 "101"
- [3,4,5] ，从 "001101" 得到 "101"
没有别的合法选择，所以总共有 6 种方法。
示例 2：

输入：s = "11100"
输出：0
解释：没有任何符合题意的选择。
 

提示：

3 <= s.length <= 105
s[i] 要么是 '0' ，要么是 '1' 。

方法1：前缀和

• 本题类似926

       public long numberOfWays(String s) {
            int n = s.length();
            long[] one = new long[n + 1];//记录'1'的数量，剩下的便是'0'的数量
            for (int i = 1; i <= n; i++) {
                one[i] = one[i - 1] + (s.charAt(i - 1) == '1' ? 1 : 0);
            }
            long res = 0;
            for (int i = 0; i < n; i++) {
                char c = s.charAt(i);
                if (c == '0') {
                    //形成101这样的结构
                    long a = one[i + 1], b = one[n] - one[i + 1];
                    res += a * b;
                } else {
                    //形成010这样的结构
                    long a = i + 1 - one[i + 1], b = n - i - (one[n] - one[i]);
                    res += a * b;
                }
            }
            return res;
        }

• 另，不用可以统计前缀和

        public long numberOfWays(String s) {
            long tot_zero = 0, cur_zero = 0;//统计所有'0'的数量 当前所有'0'的数量
            int n = s.length();
            for (int i = 0; i < n; i++) {
                if (s.charAt(i) == '0') tot_zero++;
            }
            long res = 0;
            for (int i = 0; i < n; i++) {
                char c = s.charAt(i);
                if (c == '1') {//010
                    res += cur_zero * (tot_zero - cur_zero);
                } else {//101
                    long cur_one = i - cur_zero;
                    // n - 所有的0 剩下所有的1 再减去当前的1 得到剩下的1 
                    res += cur_one * (n - tot_zero - cur_one);
                    cur_zero++;
                }
            }
            return res;

        }