codeforces 789B. Masha and geometric depression

题目链接：http://codeforces.com/contest/789/problem/B

题意：给你b1，q，l，m。b1是等比数列第一项，q是公比，l是限制条件，如果某一项的绝对值大于l就结束等比，如果某一项等于ai就不计入数量。现在问你这个数列可以列到多少个

分析：就根据条件写即可，我看到题一开始考虑的就是q=0，q=1，q=-1，以及b1=0.这种情况才有可能出现无限多。其他情况模拟即可。

AC代码：

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

map<long long,int>mp;
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    long long b,q,l,m;
    cin>>b>>q>>l>>m;
    long long x;
    for(int i=1;i<=m;i++){
        cin>>x;
        mp[x]=1;
    }
    if(q==0){
        if(mp[0]!=1){
            if(abs(b)<=l) cout<<"inf"<<endl;
            else cout<<0<<endl;
        }
        else {
            if(mp[b]!=1&&abs(b)<=l) cout<<1<<endl;
            else cout<<0<<endl;
        }
    }
    else if(q==1||q==-1){
        if(mp[b]!=1){
            if(abs(b)<=l) cout<<"inf"<<endl;
            else cout<<0<<endl;
        }
        else if(mp[b*q]!=1){
            if(abs(b*q)<=l) cout<<"inf"<<endl;
            else cout<<0<<endl;
        }
        else cout<<0<<endl;
    }
    else {
        int result=0;
        if(b==0) {
            if(mp[b]!=1){
                cout<<"inf"<<endl;
            }
            else cout<<0<<endl;
        }
        else {
            while(abs(b)<=l){
                if(mp[b]!=1){
                result++;
                }
                b=b*q;
            }
            cout<<result<<endl;
        }

    }

return 0;
}