F2. Promising String (hard version)---经典前缀和树状数组

暴力版本(easy version)

思路：
首先要想满足条件，则子区间内-号数量必须大于等于+号数量
其次-号数量减去 +数量必须是3的整数倍
因此用前缀和记录符号数量遇-就加1否则减1
暴力枚举左右端点，然后判断该区间能否满足条件即可

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<map>
using namespace std;
#define int long long
const int N = 5010;
int s[N];
void solve()
{
	int n;cin>>n;s[0]=0;
	for(int i=1;i<=n;i++)
	{
	    char op;cin>>op;
		s[i]=s[i-1];
		if(op=='+')s[i]--;
		if(op=='-')s[i]++;
	}
	int res=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=i;j++)
		{
			int t=s[i]-s[j-1];
			if(t>=0&&t%3==0)res++;
		}
	}
	cout<<res<<'\n';
}
signed main()
{
    int T;cin>>T;
    while(T--)solve();
}

经典前缀和树状数组题型(hard version)

思路：由第一题可知暴力必然超时
我们需要对代码进行优化
首先对于一个区间做右端点必须满足$s[r]-s[l-1]>=0$还需要满足$(s[r]-s[l-1])mod3=0$
即在mod3的情况下$s[r]=s[l-1]$
可以想到用模数为0，1，2的树状数组计算以r为右端点的符合情况的左端点数量就好了
需要注意的是必须满足$s[r]-s[l-1]>=0$，所以查询时在同模一个数的情况下查询小于s[r]的数
可能出现负数，所以需要进行一次映射，使记录符号个数的前缀和数组全都大于1

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<map>
using namespace std;
#define int long long
const int N = 400010;
int s[N];
int n;
struct BIT
{
	int tr[N];
	int lowbit(int x)
    {
        return x&-x;
    }
    void add(int x,int v)
    {
        for(int i=x;i<N;i+=lowbit(i))tr[i]+=v;
    }
    int query(int x)
    {
        int res=0;
        for(int i=x;i;i-=lowbit(i))res+=tr[i];
        return res;
    }
	void init(int n)
	{
		for(int i=0;i<n+2;i++)tr[i]=0;
	}
	
}tr[3];
 
void solve()
{
	cin>>n;
	int minv=0;s[0]=0;
	for(int i=1;i<=n;i++)
	{
	    char op;cin>>op;
		s[i]=s[i-1];
		if(op=='+')s[i]--;
		if(op=='-')s[i]++;
		minv=min(minv,s[i]);
	}
	for(int i=0;i<=n;i++)s[i]-=minv-1;
	for(int i=0;i<3;i++)tr[i].init(n-minv+1);
	int res=0;
	for(int i=0;i<=n;i++)
	{
		int t=s[i]%3;
		res+=tr[t].query(s[i]);
		tr[t].add(s[i],1);
//		for(int j=1;j<=i;j++)
//		{
//			int t=s[i]-s[j-1];
//			if(t>=0&&t%3==0)res++;
//			
//		}
	}
	cout<<res<<'\n';
}
 
signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    int T;cin>>T;
    while(T--)solve();
}

合适数对—同类型题

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define int long long
const int N = 400010;
int s[N];
int tr[N];
int n,m;
vector<int>nums;
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int v)
{
    for(int i=x;i<N;i+=lowbit(i))tr[i]+=v;
}
int query(int x)
{
    int res=0;
    for(int i=x;i>=1;i-=lowbit(i))res+=tr[i];
    return res;
}
int get(int x)
{
    return lower_bound(nums.begin(),nums.end(),x)-nums.begin()+1;
}
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>s[i],s[i]+=s[i-1];
    for(int i=0;i<=n;i++)
    {
        nums.push_back(s[i]);
        nums.push_back(s[i]-m);
    }
    sort(nums.begin(),nums.end());
    nums.erase(unique(nums.begin(),nums.end()),nums.end());
    int res=0;
    for(int i=0;i<=n;i++)
    {
        //cout<<"res +  "<<query(N-1)-query(get(s[i]-m))<<endl;
        //cout<<s[i]<<endl;
        
        res+=query(N-1)-query(get(s[i]-m));
        add(get(s[i]),1);
    }
    cout<<res<<'\n';
}