Q481 神奇字符串

神奇的字符串 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;
    }
}