Codeforces Round #780 (Div. 3) F2. Promising String (hard version)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define int long long 
const int N=4e5+10;
int ADD;
struct node{
    int n,a[N];
    void init(int _n){n=_n;for(int i=1;i<=n;i++) a[i]=0;}
    int lowbit(int x) {return x&(-x);}
    void add(int pos,int val){a[pos]+=val;while((pos+=lowbit(pos))<=n){a[pos]+=val;}}
    int getsum(int pos){int sum=a[pos];while((pos-=lowbit(pos))>0){sum+=a[pos];};return sum;}
}bit[4];
int t,n;string s;
int32_t main(){
    // cout<<(-2%3)<<endl;
    scanf("%lld",&t);
    while(t--){
        scanf("%lld",&n);cin>>s;int now=0,res=0;
        ADD=n+1;
        for(int i=0;i<=2;i++) bit[i].init(ADD<<1);
        bit[0].add(ADD,1);
        for(int i=0;i<n;i++){
            now+=(s[i]=='-'?-1:1);
            res+=bit[(now%3+3)%3].getsum(ADD<<1)-bit[(now%3+3)%3].getsum(now+ADD-1);
            bit[(now%3+3)%3].add(now+ADD,1);
            // cout<<(now%3+3)%3<<" "<<res<<endl;
        }
        cout<<res<<endl;
        
    }
}