判断元素是否存在

题目来源：http://ybt.ssoier.cn:8088/problem_show.php?pid=1211

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 7093 通过数: 2734
【题目描述】
有一个集合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
【来源】
No

思路：递归，复杂度log级别，不会TLE。注意递归边界，包含小于k的那些。

#include <bits/stdc++.h>
using namespace std;

int k, x;

bool belongto(int a)
{
    if (a < k)
    {
        return false;
    }
    if (a == k)
    {
        return true;
    }
    a -= 1;
    if (a % 2 == 0)
    {
        if (a % 3 != 0)
        {
            return belongto(a / 2);
        }
        else
        {
            return belongto(a / 2) || belongto(a / 3);
        }
    }
    else
    {
        if (a % 3 != 0)
        {
            return false;
        }
        else
        {
            return belongto(a / 3);
        }
    }
}

int main()
{
    scanf("%d", &k);
    getchar();
    scanf("%d", &x);
    bool ans = belongto(x);
    printf(ans ? "YES\n" : "NO\n");
    // system("pause");
}