LeetCode 6035. 选择建筑的方案数

一、题目

1、题目描述

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

2、基础框架

• C语言给出的基础框架代码如下：

long long numberOfWays(char * s){

}

3、原题链接

LeetCode 6035. 选择建筑的方案数

二、解题报告

1、思路分析

  $(1)$ 如果说要求的是方案数，大概率是动态规划；如果求某一种方案，大概率是搜索。那么这题，大概率是动态规划。
  $(2)$ 如果用 $f[x][i]$ 表示以第 $i$ 个字符结尾，长度为 $x$ 的方案数。$f[2][7]$ 表示以第 $7$ 个字符结尾，长度为 $2$ 的方案数。
  $(3)$ 假设第 $i$ 个字符为 $s[i]$，则状态转移方程：
$$\begin{gathered} f[1][i]=1 \\ f[2][i]=\sum _{j=0}^{i-1}(s[j]==s[i]?0:1) \\ f[3][i]=\sum _{j=0}^{i-1}(s[j]==s[i]?0:f[2][j]) \end{gathered}$$

  $(4)$ 对于 $f[2][i]$，我们用两个数组 $sum[0][i]$ 和 $sum[1][i]$ 分别记录以第 $i$ 个元素结尾的前缀和。
  $(5)$ 对于 $f[3][i]$ 也同样处理。

2、时间复杂度

   最坏时间复杂度 $O(n)$ 。

3、代码详解

#define ll long long
#define maxn 100005

ll sum[2][maxn];
ll f[4][maxn];

long long numberOfWays(char * s){
    ll ret = 0;
    int n = strlen(s);
    int x, i, j;

    for(j = 1; j <= 3; ++j) {
        if(j == 1) {
            for(i = 0; i < n; ++i) {
                f[j][i] = 1;
            }
            continue;
        }
        for(x = 0; x < 2; ++x) {
            for(i = 0; i < n; ++i) {
                sum[x][i] = (s[i] - '0') == x ? f[j-1][i] : 0;
                if(i) {
                    sum[x][i] += sum[x][i-1];
                }
            }
        }
        for(i = 0; i < j-1; ++i) {
            f[j][i] = 0;
        }
        for(i = j-1; i < n; ++i) {
            x = s[i] - '0';
            f[j][i] = sum[x^1][i-1];
            if(j == 3)
                ret += f[j][i];
        }
    }

    return ret;

}

---

三、本题小知识

  如果说要求的是方案数，大概率是动态规划；如果求某一种方案，大概率是搜索。那么这题，大概率是动态规划。

---

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

🌌《算法入门指引》🌌
  如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

---

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜