神奇字符串

/*
思路：
虽然Java Stream不太适合处理这个问题，但我们可以先生成字符数组，然后使用Stream统计 '1' 的数量。
*/
 
import java.util.stream.IntStream;
 
public class MagicStringStream {
    public int magicalString(int n) {
        if (n == 0) return 0;
        if (n <= 3) return 1;
        int[] s = new int[n + 1];
        s[0] = 1; s[1] = 2; s[2] = 2;
        int head = 2, tail = 3, num = 1;
        while (tail < n) {
            for (int i = 0; i < s[head]; i++) {
                s[tail] = num;
                tail++;
            }
            num = num == 1 ? 2 : 1;
            head++;
        }
        return (int) IntStream.range(0, n).filter(i -> s[i] == 1).count();
    }
}