神奇的字符串 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。
class Solution {
public int magicalString(int n) {
if (n <= 3)
return n > 1 ? 1 : n;
int c = 2;
int idx = 2;
int k = 3;
int result = 1;
short[] nums = new short[n + 1];
nums[0] = (short)1;
nums[1] = (short)2;
nums[2] = (short)2;
while (k < n) {
c = c == 2 ? 1 : 2;
if (nums[idx] == 1) {
nums[k++] = (short)c;
if (c == 1)
result++;
} else {
nums[k++] = (short)c;
nums[k++] = (short)c;
if (c == 1) {
result += (k <= n ? 2 : 1);
}
}
idx++;
}
return result;
}
}
原文:https://www.cnblogs.com/WeichengDDD/p/10738927.html