[数据结构与算法]LeetCode 481. 神奇字符串 -- 找规律

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 的数目。

示例 1：

输入：n = 6
输出：3
解释：神奇字符串 s 的前 6 个元素是 “122112”，它包含三个 1，因此返回 3 。

示例 2：

输入：n = 1
输出：1

提示：

1 <= n <= 105

题解

找规律完事，直接看代码就行了。

AC代码

class Solution {
public:
    string s;
    int magicalString(int n) 
    {
        if(n<=3)return 1;
        s = "122";
        int pos = 2;
        while(s.length()<n)
        {
            if(s[pos]=='2')
            {
                if(s[s.length()-1] == '1')
                s += "22";
                else
                s += "11";
            }
            else
            {
                if(s[s.length()-1] == '1')
                s += "2";
                else
                s += "1";
            }
            pos ++;
        }
        int ans = 0;
        for(int i=0;i<n;i++)
        if(s[i]=='1')ans++;
        return ans;
    }
};