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");
    }
}