神奇的字符串 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' 的数目。
注意:N 不会超过 100,000。
示例:
输入:6 输出:3 解释:神奇字符串 S 的前 6 个元素是 “12211”,它包含三个 1,因此返回 3。
思路如下:总结规律,下标i为偶数时,该数所对应的repo[i]个数值应当为1;
i为奇数时,该数所对应的repo[i]个数值应为2;
且每次添加,都可以用3-1=2,3-2=1这样来模拟1,2,1,2。。。的变化。
以下为代码:
class Solution {
public:
int magicalString(int n) {
if(n<1)
return 0;
if(n<=3)
return 1;
vector<int> repo;
repo.push_back(1);
repo.push_back(2);
repo.push_back(2);
long total=3;
int num;
long coun=1;
for(int i=2;total<n;i++){
if(i%2==0){
num=1;
coun+=repo[i];
}
else num=2;
for(int k=0;k<repo[i];k++){
repo.push_back(num);
}
total+=repo[i];
}
if(num==1&&total-n==1)
return coun-1;
return coun;
}
};