题目描述:
神奇字符串 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,以此类推。
由于前两组数字已知,我们初始化字符串 s 为 122,然后从第三组开始构造,第三组数字是由字符串 s 的第三个数字(下标 i=2)得到,因此我们此时将指针 i 指向字符串 s 的第三个数字 2。
1 2 2
^
i
指针 i 指向的数字为 2,表示第三组的数字会出现两次,并且,由于前一组数字为 2,组之间数字交替出现,因此第三组数字为两个 1,即 11。构造后,指针 ii移动到下一个位置,即指向字符串 s 的第四个数字 1。
1 2 2 1 1
^
i
这时候指针 i 指向的数字为 1,表示第四组的数字会出现一次,并且,由于前一组数字为 1,组之间数字交替出现,因此第四组数字为一个 2,即 2。构造后,指针 i 移动到下一个位置,即指向字符串 s 的第五个数字 1。
1 2 2 1 1 2
^
i
我们按照这个规则,依次模拟构造过程,直到字符串 s 的长度大于等于 n。
代码:
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);
}
};
提交结果:
![](https://i-blog.csdnimg.cn/blog_migrate/97602421faea049f71d04d4961a02b49.png)