CF1542 B. Plus and Multiply

Problem - 1542B - Codeforces

 

题意：

若x在集合里，则x*a和x+b都在集合里

初始集合里只有1，给出n，问n是否在集合里

 

x先乘a再加b得到x*a+b

x先加b再乘a得到x*a+b*a

所以x先乘a再加a次b等价于x先加b再乘a

所有的集合里的数都可以表示为x*a^i+b*j

若a!=1，枚举i

若a=1，直接解

 

#include<bits/stdc++.h>
 
using namespace std;
 
int main()
{
    int T,n,a,b;
    long long aa;
    bool tag;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&a,&b);
        tag=false;
        if(a!=1)
        {
            aa=1;
            for(int i=1;aa<=n;aa*=a)
                if(!((n-aa)%b))
                {
                    tag=true;
                    break;
                }
        }
        else if(!((n-1)%b)) tag=true;
        puts(tag ? "YES" : "NO");
    }
}