选择建筑的方案数

题目
给你一个下标从 0 开始的二进制字符串 s ，它表示一条街沿途的建筑类型，其中：

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-ways-to-select-buildings

思路

前缀和

代码

class Solution {
public:
    long long numberOfWays(string s) {

        // 考察前缀和


        int n=s.size();

        vector<int> leftOne(n),rightOne(n),leftZero(n),rightZero(n);

        for(int i=1;i<n;i++){
            
            if(s[i-1]=='0'){
            leftZero[i]=leftZero[i-1]+1;
            leftOne[i]=leftOne[i-1];
            }else{
            leftOne[i]=leftOne[i-1]+1;
            leftZero[i]=leftZero[i-1];
            }
        }

        for(int i=n-2;i>=0;i--){
            if(s[i+1]=='0'){
            rightZero[i]=rightZero[i+1]+1;
            rightOne[i]=rightOne[i+1];
            }else{
                rightOne[i]=rightOne[i+1]+1;
                rightZero[i]=rightZero[i+1];
            }
        }

        long long ans=0;

        for(int i=1;i<n-1;i++){

            if(s[i]=='0')
            ans+=leftOne[i]*rightOne[i];
            else
            ans+=leftZero[i]*rightZero[i];
        }

        return ans;
    }
};