SRM609 Div1 Easy MagicalStringDiv1

問題概要

>と<からなる文字列の部分列のうちi個の>の後にi個の<が来るようなものうち最大の長さ

解法

境界を決めて左の>の個数と右の<の個数のうち小さい方を二倍したものを更新していく.

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(ll i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(ll i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
class MagicalStringDiv1{
public:
    int getLongest(string st){
        int n=st.size();
        vi a(n);
        REP(i,n){
            if(st[i]=='>'){
                a[i]++;
            }
            if(i){
                a[i]+=a[i-1];
            }
        }
        vi b(n);
        REPR(i,n-1){
            if(st[i]=='<'){
                b[i]++;
            }
            if(i){
                b[i-1]+=b[i];
            }
        }
        int ans=0;
        REP(i,n-1){
            chmax(ans,min(a[i],b[i+1])*2);
        }
        return ans;
    }
};