leetcode481

先貼一下代碼

class Solution {
public:
    int magicalString(int n) {
        vector<int>tmp;
        tmp.push_back(1);
        tmp.push_back(2);
        tmp.push_back(2);
        
        int v = 2; //從座標2開始製造字符串
        int c = 0; //判斷當前應該加的是1還是2
        int ans = n>0 ? 1 : 0;
        
        while(v < n && n>3)
        {
            if(!c)  //如果當前應該加的是1
            {
                for(int i = 0;i<tmp[v];i++)   //tmp[v]是應該加的數的個數
                {
                    
                    tmp.push_back(1);
                    ans++;
                    if(tmp.size() == n) return ans;
                }
            }
            else
            {
                for(int i = 0;i<tmp[v];i++)
                {
                    tmp.push_back(2);
                    if(tmp.size() == n) return ans;
                }
            }
            v++;       
            c = (c+1)%2;
        }
        return ans;
    }
};

其實在座標2時就可以開始製造這個字符串了，是有規律的，設定指針，判斷當前添加1時應該添加幾個1，添加2時應該添加幾個2。然後往後移動一格，之前加的是1就換成2，以此類推即可。在製造字符串時同時記下1的個數，一旦字符串長度到達n就退出。