Leetcode--Java--6035. 选择建筑的方案数 (周赛T3)

题目描述

给你一个下标从 0 开始的二进制字符串 s ，它表示一条街沿途的建筑类型，其中：
s[i] = ‘0’ 表示第 i 栋建筑是一栋办公楼，
s[i] = ‘1’ 表示第 i 栋建筑是一间餐厅。
作为市政厅的官员，你需要随机 选择 3 栋建筑。然而，为了确保多样性，选出来的 3 栋建筑 相邻 的两栋不能是同一类型。
比方说，给你 s = “001101” ，我们不能选择第 1 ，3 和 5 栋建筑，因为得到的子序列是 “011” ，有相邻两栋建筑是同一类型，所以 不合 题意。
请你返回可以选择 3 栋建筑的 有效方案数 。

样例描述

输入：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 种方法。

思路

双向遍历统计 + 前缀和思想

1. 显然只有101和010两种情况。对于每个位置的1，统计其左边0的个数以及右边0的个数，相乘就是010方案数。同理，对于每个位置的0，统计左右两边1的个数
2. 具体用一个count数组来统计。不需要开两个，只要分别针对0/1统计即可。例如对于0，count统计的就是右边的1的个数，针对1count统计右边0的个数。 注意这里统计右边应该逆序遍历，从右往左依次递推
3. 在从左往右遍历，统计对于不同位置左边的0或者1的个数，并将成绩累加的结果中

代码

class Solution {
    public long numberOfWays(String s) {
        long ans = 0;
        char ch[] = s.toCharArray();
        int n = ch.length;
        int count[] = new int[n];//count[i]表示位置i右侧的0/1的数量
        int zeros = 0, ones = 0; //统计0和1的数量
        //从右往左统计当前位置右边0/1的数量
        for (int i = n - 1; i >= 0; i -- ) {
            if (ch[i] == '1') {
                count[i] = zeros;
            } else {
                count[i] = ones;
            }
            zeros += (ch[i] == '0' ? 1 : 0);
            ones += (ch[i] == '1' ? 1 : 0);
        }
        //清空，准备逐步统计左边的0和1的个数
        zeros = 0;
        ones = 0;
        //从左往右统计左侧，并乘积后累加求和
        for (int i = 0; i < n; i ++ ) {
            if (ch[i] == '1') {
                ans += zeros * count[i];
            } else {
                ans += ones * count[i];
            }
            zeros += (ch[i] == '0' ? 1 : 0);
            ones += (ch[i] == '1' ? 1 : 0);
        }
        return ans;
    }
}