腾讯音乐娱乐集团2024校园招聘-前端开发笔试（II）

小红的二叉树计数

题目描述

小红定义一个二叉树为“好二叉树”，当且仅当该二叉树所有节点的孩子数量为偶数（$0$ 或者 $2$）。小红想知道，$n$ 个节点组成的好二叉树，共有多少种不同的形态？答案请对 $10^9+7$ 取模。

数据范围

• $1\le n\le 3000$

解题思路

1. 所有好二叉树的节点的数量均为奇数；
2. 一棵二叉树的形态由其左子树和右子树的形态决定。

代码实现

int cntOfTrees(int n) {
    const int N = 3e3 + 5, MOD = 1e9 + 7;
    long long dp[N] = {0};
    // 节点个数为偶数时无法构成好二叉树
    if (!(n & 1)) return 0;
    dp[1] = dp[3] = 1;
    for (int i = 5; i <= n; i += 2) {
        for (int j = 1, k = i - 1; j < k; j += 2) {
            dp[i] += (dp[j] * dp[k - j]) % MOD;
            dp[i] %= MOD;
        }
    }
    return (int) dp[n];
}

时间复杂度：$O(n^2)$

空间复杂度：$O(n^2)$

小红的可爱串

题目描述

小红定义一个字符串是可爱串，当且仅当该字符串包含子序列 $red$，且不包含子串 $red$。

我们定义子序列为字符串中可以不连续的一段，而子串则必须连续。例如 $rderd$ 包含子序列 $red$，且不包含子串 $red$，因此该字符串为可爱串。

小红想知道，长度为 $n$ 的、仅由 $r$、$e$、$d$ 三种字母组成的字符串中，有多少是可爱串？答案请对 $10^9+7$ 取模。

数据范围

• $1\le n\le 10^5$

解题思路

1. 包含子串 $red$ 的字符串的集合 ∈ 包含子序列 $red$ 的字符串集合。

2. 可爱串的数量 = 包含子序列 $red$ 的字符串的数量 - 包含子串 $red$ 的字符串的数量 。

3. 如何构造包含子串 $red$ 的字符串（记字符串为 $s$） ？

   • 令 $f[i]$ = 长度为 $i$ 的字符串中包含子串 $red$ 的字符串的数量。
   • 分为两种情况
     • 若 $s[i]$ 为 $r$ 或 $e$，则 $s[i]$ 必定不影响 $f[i]$，$f[i]=f[i-1]+f[i-1]$。
     • 若 $s[i]$ 为 $d$
       • 若 $s[i] $ 不影响 $f[i]$，即 $s[\mathrm{1..}i-1]$ 已包含子串 $red$，则$ f[i] = f[i-1]$。
       • 若 $s[i] $ 影响 $f[i]$，即 $s[\mathrm{1..}i-3]$ 不含子串 $red$，但 $s[i-\mathrm{2..}i-1]=re$，加上 $s[i]$ 即可构成子串 $red$，则$ f[i] = 3^{(i-3)}-f[i-3]$。
   • 综上，$f[i]=3\times f[i-1]+3^{(i-3)}-f[i-3]$。

4. 如何构造包含子序列 $red$ 的字符串（记字符串为 s）？

   • 令 $g[i]$ = 长度为 $i$ 的字符串中包含子序列 $red$ 的字符串的数量。
   • 分为两种情况
     • 若 $s[i]$ 为 $r$ 或 $e$，则 $s[i]$ 必定不影响 $g[i]$，$g[i]=g[i-1]+g[i-1]$。
     • 若 $s[i]$ 为 $d$
       • 若 $s[i] $ 不影响 $g[i]$，即 $s[\mathrm{1..}i-1]$ 已包含子序列 $red$，则 $g[i]=g[i-1]$。
       • 若 $s[i] $ 影响 $g[i]$，即 $s[\mathrm{1..}i-i]$ 不含子序列 $red$，但含子序列 $re$，加上 $s[i]$ 即可构成子序列 $red$，令 $h[i]$ = 长度为 $i$ 的字符串中不含子序列 $red$但含子序列 $re$ 的字符串的数量，则 $g[i]=h[i-1]$。
     • 综上所述，$g[i]=3\times g[i-1]+h[i-1]$。

5. 如何构造不含子序列 $red$且含子序列 $re$ 的字符串，即如何计算 $f[i]$（记字符串为 s）？

   • 令 $h[i]$ = 长度为 $i$ 的字符串中不含子序列 $red$ 但含子序列 $re$ 的字符串的数量（第4点中已进行如此假设）。

   • $s[i]$ 是否能为 $d$？包含子序列 $re$、不包含子序列 $red$、$s[i]=d$，满足这三个条件的字符串不存在，故不考虑 $s[i]=d$ 的情况。

   • 分为两种情况

     • 若 $s[i]$ 为 $r$，则 $s[i]$ 必定不影响 $h[i]$，$h[i]=h[i-1]$。

     • 若 $s[i]$ 为 $e$

       • 若 $s[i]$ 不影响 $h[i]$，即 $s[\mathrm{1..}i-1]$ 已包含子序列 $re$ 且不含子序列 $red$，则 $h[i]=h[i-1]$。

       • 若 $s[i]$ 影响 $h[i]$，即 $s[\mathrm{1..}i-1]$ 不满足（已包含子序列 $re$ 且不含子序列 $red$），但含有子序列 $r$，加上 $s[i]$ 即可构成子序列 $re$。

         令条件 $A$ = 不满足（已包含子序列 $re$ 且不含子序列 $red$）但含有子序列 $r$，设满足条件 $A$ 的字符串为 $t$，下面考虑如何构造 $t$。

         $t$ 的长度为 $i-1$，$t$ 至少包含一个子序列 $r$，手动将一个子序列 $r$ 填入 $t$，记作 $R$，并规定手动填入的 $R$ 的左边空位不能填入 $r$，$R$ 有 $i-1$ 个位置可以选择，以 $R$ 为界

         • $R$ 左边的空位只可填入 $e$ 或 $d$，这是因为，为了防止重复计算，人为规定 $R$ 的左边空位不能填 $r$。
         • $R$ 右边的空位只可填入 $r$ 或 $d$，这是因为，$R$ 右边空位若填入 $e$，则 $s[\mathrm{1..}i-1]$ 则含有子序列 $re$，$s[i]$ 对 $h[i]$ 无影响，这与前置条件相矛盾，故$R$ 右边空位不可以填入 $e$。

         经过分析，$R$ 左边和右边空位均只可以填入两种字符，共有 $i-2$ 个空位，共有 $2^{(i-2)}$ 种填充方式，又因为 $R$ 有 $i-1$ 中填充方式，故 $t$ 的数量有 $(i-1) \times 2^{(i-2)} $，也即 $h[i]=(i-1)\times 2^{(i-2)}$。

     • 综上所述，$h[i]=2\times h[i-1]+(i-1)\times 2^{(i-2)}$。

6. 综上所述，长度为 $n$ 的可爱串的数量为 $g[n]-f[n]$。

代码实现

typedef long long ll;
const int N = 1e5 + 5, MOD = 1e9 + 7;
ll f[N], g[N], h[N];

// 快速幂
ll ksm(ll x, ll n) {
    ll res = 1 % MOD;
    while (n) {
        if (n & 1)res = res * x % MOD;
        x = x * x % MOD;
        n >>= 1;
    }
    return res;
}

int kawaiiStrings(int n) {
    // f[i] = 长度为 i 的字符串中包含子串 red 的字符串的数量
    for (int i = 3; i <= n; i++)
        f[i] = (f[i - 1] * 3 + ksm(3, i - 3) - f[i - 3]) % MOD;
    // h[i] = 长度为 i 的字符串中不含子序列 red 但含子序列 re 的字符串的数量
    for (int i = 2; i <= n; i++)
        h[i] = (h[i - 1] * 2 + (i - 1) * ksm(2, i - 2)) % MOD;
    // g[i] = 长度为 i 的字符串中包含子序列 red 的字符串
    for (int i = 3; i <= n; i++)
        g[i] = (g[i - 1] * 3 + h[i - 1]) % MOD;
    return int((g[n] - f[n] + MOD) % MOD);
}

时间复杂度：$O(n)$

空间复杂度：$O(n)$

小红的元素乘积

题目描述

小红定义一个数为“完美数”，当且仅当该数仅有一个非零数字。例如 $5000,4,1,10,200$ 都是完美数。小红拿到了一个大小为 $n$ 的数组，她希望选择两个元素，满足它们的乘积为完美数。小红想知道，共有多少种不同的取法？

数据范围

• $1\le n\le 2000$
• $1\le a_i\le 10^9$

解题思路

枚举判断即可。

代码实现

typedef long long ll;

bool check(ll x) {
    if (x < 10)return true;
    string s = to_string(x);
    for (int i = 1; i < s.size(); i++)
        if (s[i] != '0')return false;
    return true;
}

int perfectPair(vector<int> &arr) {
    int n = arr.size(), res = 0;
    if (n < 2)return 0;
    for (int i = 0; i < n; i++) {
        for (int j = i + 1; j < n; j++) {
            if (check((ll) arr[i] * arr[j]))res++;
        }
    }
    return res;
}

时间复杂度：$O(n^2)$

空间复杂度：$O(1)$

END

题目来源：腾讯音乐娱乐集团2024校园招聘-前端开发笔试（II）

文章声明：题目来源 牛客 平台，如有侵权，请联系删除！

文章文档：公众号 字节幺零二四 回复关键字可获取本文文档。