Leetcode/src/CountandSay/program.cpp at master · robling/Leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include "../include/pre.h"

string countAndSayHelper(const string& s)
{
    std::stringstream ss;
    char curCh = 0;
    int curStack = 0;
    for (auto ch : s)
    {
        if (ch != curCh)
        {
            if (curStack >= 1) ss << curStack << curCh;
            curCh = ch;
            curStack = 1;
        }
        else
        {
            curStack++;
        }
    }
    if (curStack >= 1) ss << curStack << curCh;
    else if (curStack == 1) ss << curCh;
    return ss.str();
}

string countAndSay(int n)
{
    string rst("1");
    while(--n)
    {
        rst = countAndSayHelper(rst);
    }
    return rst;
}

int main()
{
    int i = 0;
    while(++i != 10)
    {
        cout << i << "\t:   " << countAndSay(i) << endl;
    }
    return 0;
}