Submission #3772580

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
#define ALL(obj) (obj).begin(),(obj).end()
#define sortall(obj) sort(ALL(obj))
#define revall(obj) reverse(ALL(obj))
#define SortRevAll(obj) revall(sortall(obj))
typedef string str;
typedef long long ll;
const int mod=1e9+7;
const int INF=1e9;
const ll LINF=1e18;
#define V vector
#define P pair
#define F first()
#define S second()
#define PB(a) push_back(a)
#define IN(a) insert(a)
#define ins(a) int a;cin >> a
#define lls(a) ll a;cin >> a
#define in(a) cin >> a
#define sts(a) string (a);cin >> a
#define For(i,m,n) for(int i=m;i<n;i++)
#define rep(i,n) For(i,0,n)
#define Forr(i,m,n) for(int i=n-1;i>=m;i--)
#define repr(i,n) Forr(i,0,n)
#define print(a) cout << a << endl
#define Endl cout << endl
#define YES(n) cout << ((n)? "YES" : "NO") << endl
#define Yes(n) cout << ((n)? "Yes" : "No") << endl
#define POSSIBLE(n) cout << ((n) ? "POSSIBLE" : "IMPOSSIBLE" ) << endl
#define Possible(n) cout << ((n) ? "Possible" : "Impossible"  ) << endl

int main()
{
    sts(s);
    ins(x);ins(y);
    V<int>cntx,cnty;
    int sumx=0,sumy=0;
    bool ansx=false,ansy=false;
    
    int oldf=0,cnt=0;
    s.PB('T');
    rep(i,s.size()){
        if(s.at(i)=='T'){
            int plus=i-oldf-1;
            if(cnt==0){
                cntx.PB(i);
            }else if(cnt%2){
                cnty.PB(plus);
            }else{
                cntx.PB(plus);
            }
            oldf=i;
            cnt++;
        }
        
    }
    V<V<int> > rx(cntx.size()),ry(cnty.size());
    rep(i,cntx.size())rx.at(i).PB(-1);
    rep(i,cnty.size())ry.at(i).PB(-1);
    rep(i,cntx.size()){
        rx[i][0]=cntx.at(i);
        sumx+=cntx.at(i);
        For(j,1,i+1){
            rx[i][j]=rx[i][0]+rx[i-1][j-1];
        }
    }
    rep(i,cnty.size()){
        ry[i][0]=cnty.at(i);
        sumy+=cnty.at(i);
        For(j,1,i+1){
            ry[i][j]=ry[i][0]+ry[i-1][j-1];
        }
    }
    if(sumx==x)ansx=true;
    if(sumy==y)ansy=true;
    rep(i,cntx.size()){
        rep(j,i+1){
            if(x==sumx-rx[i][j]*2)ansx=true;
        }
    }
    rep(i,cnty.size()){
        rep(j,i+1){
            if(y==sumy-ry[i][j]*2)ansy=true;
        }
    }
    Yes(ansx & ansy);
    return 0;
}

Submission Info

Judge Result