题目描述:
神奇字符串 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
思路;
根据题目,我们得知,字符串s的每一组数字都可以由字符串s自身的数字得到。
字符串s前两组数字为1和22,是由字符串s的第一个数字1和第二个数字2得到的,并且第一组数字只包含1,第二组数字只包含2,第三组数字只包含1,以此类推。
由于前两组数字已知,我们初始化字符串 ss 为122,然后从第三组开始构造,第三组数字是由字符串s的第三个数字(下标i=2)得到,因此我们此时将指针i指向字符串s的第三个数字2。
1 2 2
^
i
指针i指向的数字为2,表示第三组的数字会出现两次,并且,由于前一组数字为2,组之间数字交替出现,因此第三组数字为两个1,即 11。构造后,指针i移动到下一个位置,即指向字符串s的第四个数字1。
1 2 2 1 1
^
i
这时候指针i指向的数字为1,表示第四组的数字会出现一次,并且,由于前一组数字为1,组之间数字交替出现,因此第四组数字为一个2,即2。构造后,指针i移动到下一个位置,即指向字符串s的第五个数字1。
1 2 2 1 1 2
^
i
代码:
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);
}
};