LeetCode 481. 神奇字符串

最近刷LeetCode题目的一些思路,题目信息

神奇的字符串 只包含 '1' 和 '2',并遵守以下规则:

字符串 S 是神奇的,因为串联字符 '1' 和 '2' 的连续出现次数会生成字符串 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 作为输入,返回神奇字符串 中前 N 个数字中的 '1' 的数目。

注意:N 不会超过 100,000。

示例:

输入:6
输出:3
解释:神奇字符串 S 的前 6 个元素是 “12211”,它包含三个 1,因此返回 3。

--------------------------------------------------------------------------------------

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

public static int magicalString(int n) {
    int retData = 1;
    List<Integer> dataList = new ArrayList<>();
    dataList.add(1);
    dataList.add(2);
    dataList.add(2);
    if(n<=3){
        if(n <= 0){
            return 0;
        }
        return retData;
    }
    int currentPosition = 2;
    while (dataList.size() < n){
        int needData = dataList.get(currentPosition);
        int lastData = dataList.get(dataList.size() - 1);
        if(lastData == 1){
            for(int i= 0; i<needData; i++){
                dataList.add(2);
            }
        } else {
            for(int i= 0; i<needData; i++){
                dataList.add(1);
                retData++;
            }
        }
        currentPosition++;

    }
    if(dataList.size() > n && dataList.get(dataList.size()-1) == 1){
        retData--;
    }
    return retData;
}

猜你喜欢

转载自blog.csdn.net/sundyguo/article/details/81096152
今日推荐