有一个集合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>
using namespace std;
int pi=0;
int p(int k,int x)
{
int t=(2*k+1);
int z=(3*k+1);
if(t>100000&&z>100000) return 0;
if(x==t||x==z) {pi=1; return 1;}
else
{
p(z,x);
if(pi==1) return 1;
p(t,x);
if(pi==1) return 1;
}
}
int main()
{
int k,x;
cin>>k>>x;
if(p(k,x)) cout<<"YES";
else cout<<"NO";
return 0;
}
失败原因,漏掉了初值与x的比较,计算次数有能较多,且返回值较难控制,难以使返回值确定。
正确代码
#include <iostream>
using namespace std;
int x;
int p(int k)
{
if(k>x) return 0;
if(k==x) return 1;
if(k<x) return p(2*k+1)||p(3*k+1);
}
int main()
{
int k;
char c;
cin>>k>>c>>x;
if(p(k)) cout<<"YES";
else cout<<"NO";
return 0;
}
这样计算次数少,且返回值容易确定。