Leetcode 481.神奇字符串

神奇字符串

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

分析题目，该字符串是唯一的，只含有1和2，且字符串前后存在依赖关系，由于其实固定的，所以我们可以生成前三位，然后从第三位开始一边生成新的字符，一边继续遍历该字符串，一直到满足字符串的长度等于N，具体实现如下

 1 import java.util.ArrayList;
 2 import java.util.List;
 3 
 4 class Solution {
 5     public static int magicalString(int n) {
 6         if(n==1) return 1;
 7         int retData = 1;
 8         List<Integer> dataList = new ArrayList<>();
 9         dataList.add(1);
10         dataList.add(2);
11         dataList.add(2);
12         if(n<=3){
13             if(n <= 0){
14                 return 0;
15             }
16             return retData;
17         }
18         int currentPosition = 2;
19         while (dataList.size() < n){
20             int needData = dataList.get(currentPosition);
21             int lastData = dataList.get(dataList.size() - 1);
22             if(lastData == 1){
23                 for(int i= 0; i<needData; i++){
24                     dataList.add(2);
25                 }
26             } else {
27                 for(int i= 0; i<needData; i++){
28                     dataList.add(1);
29                     retData++;
30                 }
31             }
32             currentPosition++;
33         }
34         if(dataList.size() > n && dataList.get(dataList.size()-1) == 1){
35             retData--;
36         }
37         return retData;
38     }
39 }