481. 神奇字符串

题目：

神奇字符串 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 的数目。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/magical-string

解题思路：

使用两个指针，一个p指向数组要追加的个数，另一个指向数组尾部，用以添加1或者2。1与2是交叉出现的，所以1追加完后就转换为2，依次追加。

代码：

Java：

class Solution {
	    public int magicalString(int n) {
        int[] arr = new int[n+1];//存放神奇数组
        //p指向要追加的个数，tail指向尾部，count存放追加个数，value为1或者2,ans统计1的个数
        int p=1,tail=1,count = 2,value =1,ans = 1;
        arr[0] = value;
        while(tail<n) {
        	value ^=value; //转换1或者2
        	while(tail<n && count-- >0) {
        		arr[tail++] = value; //追加
        		if(value ==1) ans ++; //统计1的个数
        	}
        	count = arr[++p]; //追加的个数
        }
        
        return ans;
	    }
	}

C++:

class Solution {
public:
    int magicalString(int n) {
        string s = "\1\2\2"; // 值就是 1,2,2，这样就可以直接用 s[i] 当作个数
        for (int i = 2; s.length() < n; ++i)
            s += string(s[i], s.back() ^ 3); // 1^3=2, 2^3=1，这样就能在 1 和 2 之间转换
        return count(s.begin(), s.begin() + n, 1);
    }
};

Python：

class Solution:
    def magicalString(self, n: int) -> int:
        s = [1, 2, 2]
        i = 2
        while len(s) <= n:
            s += [s[-1]^3] * s[i]
            i = i+1
        return s[:n].count(1)