题目描述:
神奇的字符串 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。
这道题我一开始想的比较复杂了,一开始我想的是用两个字符串来进行,但是会发现两个长度并不是总是一个大另一个小的,因此考虑使用一个list,一个indexA来表示存储指向前面的那个数字用于增加到后面,因为每次不是1就是2,因此加上一个之后判断前面的是1还是2直接根据此就可以判断下一个是1还是2.
代码
class Solution {
public int magicalString(int n) {
if(n == 0){
return 0;
}
List<Integer> list = new ArrayList<>();
list.add(1);
list.add(2);
list.add(2);
int indexA = 2;
int k = 1;
while (list.size() < n) {
if(list.get(indexA) == 2){
list.add(k);
list.add(k);
}
if(list.get(indexA) == 1){
list.add(k);
}
k = k == 1 ? 2 : 1;
indexA ++;
}
int sum = 0;
for (int i = 0; i < n; i++) {
if(list.get(i) == 1){
sum += list.get(i);
}
}
return sum;
}
}