神奇字符串
神奇的字符串 S 只包含 ‘1’ 和 ‘2’,并遵守以下规则: 字符串 S 是神奇的,因为串联字符 ‘1’ 和 ‘2’
的连续出现次数会生成字符串 S 本身。 字符串 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’ 的数目。
示例:
输入:6
输出:3
解释:神奇字符串 S 的前 6 个元素是 “12211”,它包含三个 1,因此返回 3。
思路分析: 这道题关键是要明白字符串构造的规律。
规律‘1’,‘2’是交替添加,添加了‘1’后添加’2’,添加了’2’后添加’1’,如此循环,而添加的个数与这个字符串本身有关。
我们设下一次添加的为flag
根据遍历当前为1?2,确定在末尾添加flag的个数
class Solution {
public:
int magicalString(int n) {
if (n < 1)
{
return 0;
}
vector<int> s;
s.push_back(1);
s.push_back(2);
s.push_back(2);
int flag = 1;
for (int i = 2; i < n - 1; i++)
{
if (s[i] == 1)
{
s.push_back(flag);
}
else {
s.push_back(flag);
s.push_back(flag);
}
flag = flag == 1 ? 2 : 1;
}
int count = 0;
for (int i = 0; i < n; i++)
{
if (s.at(i) == 1)
{
count++;
}
}
return count;
}
};