解析:
求贡献。
用一个前缀和来表示当前i是否重新cur++
假设前缀和为sum
当前第i步,sum<0 那么对答案的贡献就是i 。然后对应伪代码,我们要给cur=i+1,
当前第i步,sum>=0 直接跳过
最终还需加上整个长度,因为如果前缀和都不为0的话,我们就相当于全部遍历一遍。
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+1000;
typedef long long ll;
char s[N];
int t;
int main()
{
cin>>t;
while(t--)
{
cin>>(s+1);
int n=strlen(s+1);
ll sum=0;
ll cur=0;
ll mi=0x3f3f3f3f;
int f=0;
for(int i=1;i<=n;i++)
{
if(s[i]=='-') cur--;
else cur++;
if(cur<0)
{
f=1;
sum+=i;
cur+=1;
}
}
sum+=n;
if(f==0) cout<<n<<endl;
else
cout<<sum<<endl;
}
}