LeetCode 2222. 选择建筑的方案数

文章目录

- 1. 题目
- 2. 解题

1. 题目

给你一个下标从 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 <= 10^5
s[i] 要么是 '0' ，要么是 '1' 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-ways-to-select-buildings
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

算出所有的 0 1 的个数
只有两种模式 101， 010
以每个0为中心，两侧1的个数相乘就是方案数；以每个1为中心，两侧0的个数相乘就是方案数，全部加总就是答案

class Solution:def numberOfWays(self, s: str) -> int:zero = s.count('0')one = len(s) - zeroans = 0left0 = 0  # 010 模式for i in range(len(s)):if s[i] == '0':left0 += 1else:ans += left0*(zero-left0)left1 = 0  # 101 模式for i in range(len(s)):if s[i] == '1':left1 += 1else:ans += left1*(one-left1)return ans