atcoder AGC040 A题＞＜

原题链接

题目大意：给你一串字符s，如果
si=’<’: ai si=’>’: ai>a(i+1)
求所有a相加的最小。

解题思路：模拟
我们先从前往后把所有的<都处理一遍，用a[i]来存储第i个位置的数的大小，即用个cnt标记，如果存在连续的就加1，不存在连续的就cnt为0。
然后我们又从后往前扫一遍，这个就把>转化成了<来处理。
如果s[i]‘>’，我们就用a[i]=a[i+1]+1来更新。但这个有一个特殊情况要考虑，就是如果s[i-1]’<'时，我们的a[i]=max(a[i-1],a[i+1])+1。

代码部分：

#include
#include
#include
#include
#include
#include
#include
#include
#define IOS ios::sync_with_stdio(false);cin.tie();cout.tie(0)

using namespace std;
typedef long long ll;
const int N=5e5+100;

map<char,int> q;
ll a[N];
string str;

int main()
{
     
    IOS;
    cin>>str;
    int n=str.size()+1;
    ll sum=0;
    ll cnt=0;

    for(int i=0;i<n;i++)
    {
     
        a[i]=max(a[i],cnt);
        if(str[i]=='<'&&i<str.size())
        {
     
            cnt++;
        }
        else cnt=0;
    }

    for(int i=n-1;i>=-1;i--)
    {
     
        if(i>=0&&str[i]=='>'&&str[i-1]=='<')
        {
     
            a[i]=max(a[i+1],a[i-1])+1;
            continue;
        }
        if(i>=0&&str[i]=='>')
        {
     
            a[i]=a[i+1]+1;
        }
    }

    for(int i=0;i<n;i++)
    {
     
        sum+=a[i];
    }
    cout<<sum<<endl;
    return 0;
}