LeetCode481. 神奇字符串解题思路

目录

题目

解决方案

---

题目

神奇字符串 s 仅由 '1' 和 '2' 组成，并需要遵守下面的规则：

    神奇字符串 s 的神奇之处在于，串联字符串中 '1' 和 '2' 的连续出现次数可以生成该字符串。

s 的前几个元素是 s = "1221121221221121122……" 。如果将 s 中连续的若干 1 和 2 进行分组，可以得到 "1 22 11 2 1 22 1 22 11 2 11 22 ......" 。每组中 1 或者 2 的出现次数分别是 "1 2 2 1 1 2 1 2 2 1 2 2 ......" 。上面的出现次数正是 s 自身。

给你一个整数 n ，返回在神奇字符串 s 的前 n 个数字中 1 的数目。

示例 1：

输入：n = 6
输出：3
解释：神奇字符串 s 的前 6 个元素是 “122112”，它包含三个 1，因此返回 3 。

示例 2：

输入：n = 1
输出：1

提示：

    1 <= n <= 105

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

解决方案

        如图所示，在字符串s中，1和2交替出现，且最多出现两次。所以第四个数字为1。

 

 

         由题意知第三组数字1出现的次数为第三个数字2，因此会出现两个1。第四组数字的数量又由第四个数字决定，所以是一个2。

 

class Solution {
public:
    int magicalString(int n) {
        vector<int> s = {1, 2, 2};
        for (int i = 2; s.size() < n; i++)
        {
            int pre = s.back();
            int cur = 3 - pre;
            for (int j = 0; j < s[i]; j++)
            {
                s.emplace_back(cur);
            }
        }
        return count(s.begin(), s.begin() + n, 1);
    }
};

count_if函数语法：

        count_if( *_begin() , *_end() , _pred)，其中 *_begin()是容器的头， *_end() 是容器的尾，_pred是一个用于判断的布尔值函数。