题目:
题意解析:
有一个无穷大的正整数集合 S,该集合按下面所述方法生成:
-
数字 1 在集合 S 中。
-
若数字 x 在该集合中,那么数 x×a 和数 x+b 均在集合 S 中。(其中 a 与 b 为给定常数)
现在给出数 n,a,b,请判断 n 是否在集合 S 中(此处给出的 a 与 b 就是上述集合生成方法中的 a 和 b),若在请输出 Yes
,否则输出 No
。多组数据。令数据组数为 t,那么有 1≤t≤10^5,1≤n,a,b≤10^9。
题解:
(1)这个题目一个数有两个走法,可以选择*,或者选择+。我一开始想到的是,用一个数记录sum = a^i(但是我使用的是pow(a,i)这样使用是错误的),然后判断(n-sum)%==0,如果等于0的话,就输出Yes,反之输出No。
(2)但是上面代码时间超限了,这里要考虑一个特殊情况,就是a=1的时候,你只要判断一下n%b=0就行了。
(3)最后说一下为什么使用pow(a,i)是错误的,pow(a,i)在数很大的情况下,会有误差,所以不要使用pow()函数。sqrt()函数也是一样的,在数很大的情况下,也会有误差。
其实这个我还想了一个解法,但是这个解法会出现内存超限的可能,就是用优先队列保存每一数走*和+的情况。然后不断取出队头元素,然后判断。要是有兴趣的话,可以试一下这个解法。
代码:
#include<bits/stdc++.h>
using namespace std;
int n;
int main() {
cin>>n;
long long int a,b,c;
while(n--) {
cin>>a>>b>>c;
if(b==1){
if((a-1)%c==0){
cout<<"Yes"<<endl;
}else{
cout<<"No"<<endl;
}
continue;
}
long long int sum=1;
int sign=0;
while(sum<=a){
if((a-sum)%c==0){
sign=1;
break;
}
sum*=b;
}
if(sign==0) {
cout<<"No"<<endl;
} else {
cout<<"Yes"<<endl;
}
}
return 0;
}