一本通-1211：判断元素是否存在（递归）

【题目描述】 有一个集合M是这样生成的：(1) 已知k是集合M的元素； (2) 如果y是M的元素，那么，2y+1和3y+1都是M的元素；(3) 除了上述二种情况外，没有别的数能够成为M的一个元素。

问题：任意给定k和x，请判断x是否是M的元素。这里的k是无符号整数，x 不大于 100000，如果是，则输出YES，否则，输出NO。

【输入】 输入整数 k 和 x, 逗号间隔。

【输出】 如果是，则输出 YES，否则，输出NO。

【输入样例】 0,22

【输出样例】 YES

---

#include <iostream>
#include <iomanip>
#include <vector>
#include <cmath>
#include <string>
#include <climits>
#include <cstdio>
#include <cstdlib>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <set>
#include <algorithm>

using namespace std;

bool solve(int n, int k)
{
    if (n == k) return true;

    if (n > k) return false;

    return solve(2 * n + 1, k) || solve(3 * n + 1, k);
}

int toNum(const string & s)
{
    int n = s.size();
    int res = 0;
    for (int i = 0; i < n; ++i) {
        res = res * 10 + s[i] - '0';
    }    

    return res;
}

int main()
{
    std::ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    string line; cin >> line;
    int pos = -1;
    for (size_t i = 0; i < line.size(); ++i) {
        if (line[i] == ',') {
            pos = i; break;
        }
    }

    int n = toNum(line.substr(0, pos)), k = toNum(line.substr(pos + 1));

    if (solve(n, k)) cout << "YES" << endl;
    else cout << "NO" << endl;

    return 0;
}