leetcode 选择建筑的方案数

给你一个下标从 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' 。

C++
 

class Solution {
public:
    long long numberOfWays(string s) {
        long long res=0;
        int n=s.size();
        vector<long> a(n,0);  // 01
        vector<long> b(n,0);  // 10
        long num_0=0;   //从右边起0的数量
        long num_1=0;   //从右边起1的数量
        long num_01=0;   //从右边起01的数量
        long num_10=0;   //从右边起10的数量
        for(int i=n-1;i>=0;i--) {
            if(s[i]=='0') {
                num_0++;
            } else {
                num_1++;
            }
            if(i<n-1) {
                if(s[i]=='0') {
                    num_01+=num_1;
                } else {
                    num_10+=num_0;
                }
                a[i]=num_01;
                b[i]=num_10;
            }
        }
        for(int i=0;i<n-2;i++) {
            if(s[i]=='0') {
                res+=b[i+1];
            } else {
                res+=a[i+1];
            }
        }
        return res;
    }
};