提出 #577650

---

ソースコード 拡げる

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

int main() {
  int n;
  string s;
  cin >> n >> s;
  int a[n+1],b[n+1],x=0,y=0;
  memset(a,0,sizeof(a));
  memset(b,0,sizeof(b));
  for(int i=0; i<n; i++) {
    a[i+1]+=a[i]+(s[i]=='>');
    x+=(s[i]=='>');
  }
  for(int i=n-1; i>=0; i--) {
    b[n-i]+=b[n-i-1]+(s[i]=='<');
    y+=(s[i]=='<');
  }
  int ans=0;
  for(int i=0; i<n; i++) {
    int res=0;
    int l=i+1,r=n-i;
    if(s[i]=='>') {
      if(a[l]-1>=b[r]) {
        int d=lower_bound(a,a+n,a[l]-b[r])-a;
        res+=x-a[d-1]+b[n-d];
      } else {
        int d=lower_bound(b,b+n,b[r]-a[l]+1)-b;
        res+=y-b[d-1]+a[n-d];
      }
    } else {
      if(b[r]-1>=a[l]) {
        int d=lower_bound(b,b+n,b[r]-a[l])-b;
        res+=y-b[d-1]+a[n-d];
      } else {
        int d=lower_bound(a,a+n,a[l]-b[r]+1)-a;
        res+=x-a[d-1]+b[n-d];
      }
    }
    ans=max(ans,res);
  }
  cout << ans << endl;
  return 0;
}

提出情報

ジャッジ結果