1776L - Controllers - CodeForces Solution

#include <bits/stdc++.h>

using namespace std;

typedef long double ld;
typedef long long lli;
typedef pair<lli,lli> pii;
typedef vector<lli> vi;

#define endl "\n"
#define fi first
#define si second
#define pb push_back
#define sz(s) lli(s.size())
#define all(s) begin(s), end(s)
#define print(s) cout<<s<<endl
#define fore(i,a,b) for(lli i = (a), TT=(b); i<TT; ++i)
#define _ ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);

//---------------------------------
//---------------------------------
const lli MOD = 1e9+7;
//---------------------------------
lli mcd(lli a, lli b) {return b ? mcd(b, a%b) : a;}

lli pos, neg;
lli x, n, a, b;
set<pii> answ;

int main(){_

    cin>>x;
    fore(i,0,x){
        char aux;
        cin>>aux;
        if(aux=='+') pos++;
        else neg++;
    }
    while(pos > 0 && neg > 0){
        lli aux = mcd(pos,neg);
        answ.insert({max(pos, neg)/aux, min(pos,neg)/aux});
        pos--;
        neg--;
    }
    cin>>n;
    while(n--){
        cin>>a>>b;
        lli aux = mcd(a,b);
        if( answ.count({max(a,b)/aux, min(a,b)/aux}) || answ.count({1,1}))
            print("YES");
        else print("NO");
    }
    return 0;
}