题目:
神奇字符串 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 的数目。
思路:
构造到 s 的长度达到 n 时停止。最后一次构造可能会填入两个数,导致 s 的长度为 n+1,因此最后统计 1 的个数时,应统计 s 的前 n 个数,而不是整个 s。
代码:
class Solution {
public:
int magicalString(int n) {
string s = "\1\2\2";
for (int i = 2; s.length() < n; ++i)
s += string(s[i], s.back() ^ 3);
return count(s.begin(), s.begin() + n, 1);
}
};