神奇字符串 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 <= 10^5
该题力扣
s是神奇字符串
p是s中的1,2出现次数。
class Solution {
public:
int magicalString(int n) {
if(n<4) return 1;
string s="122";
int p=2;//指针1
while(s.size()<n){
char lastCh=s.back();//指针2
if(s[p]=='1'){//p代表需要增加的字符个数
if(lastCh=='1') s.append("2");//12交替出现
else s.append("1");
}else{
if(lastCh=='1') s.append("22");
else s.append("11");
}
++p;
}
int ans=count(s.begin(),s.begin()+n,'1');//注意,s的长度或许会大于n,我们这里需要用s.begin()+n
return ans;
}
};
时间复杂度:O(n)
空间复杂度:O(n)
class Solution {
public:
int magicalString(int n) {
string s = "\1\2\2";//就是数字122这样就可以直接用s[i]当作个数
for (int i=2;s.size()<n;++i)
s+=string(s[i],s.back()^3);//第一个参数是个数,第二个参数是增加的字符
return count(s.begin(),s.begin()+n,1);
}
};
时间复杂度:O(n)
空间复杂度:O(n)