Codeforces Round #780 (Div. 3)F2 Promising String (hard version)题解 （二分+树状数组）

题目链接：F2. Promising String (hard version)

题解：我们可以发现某一段符合的条件一定是‘-’比‘‘+’多出3的整数倍，因为只要‘-’比‘+’多，就一定存在连续的’--‘，此时我们就可以将其变成’+‘，那么相对’+‘‘而言’-‘的数量就减少了3，那么就一定能使得最终’+‘和’-‘的数量相等，我们可以维护从起始点到每个位置时，每个位置上’-‘比’+‘多多少个，即为sum[i]，若某两个位置i，j（i<j）间的字符段满足条件的充要条件是：sum[j]>sum[i]&&(sum[j]-sum[i])%3==0，这就是一个经典二维偏序问题了，那具体怎么处理呢，我们可以枚举每个位置j看看有多少个i满足条件，这就是F1，但F2数据太大，我们可以利用树状数组，每次将所有sum排序，对于当前sum我们可以二分查找有多少个位置比自己小，再利用树状数组即可实现查找位置小于i并且sum也小于sum[i]的个数，因为要查找的数得满足对3取余和当前相同，所有我们可以开二维树状数组，将对3取余的结果不同的数分开存即可

不懂和细节见代码：

记得开ll(别问我怎么知道的，问就是WA45)

#include<iostream>
#include<stack>
#include<list>
#include<set>
#include<vector>
#include<algorithm>
#include<math.h>
#include<numeric>
#include<map>
#include<cstring>
#include<queue>
#include<iomanip>
#include<cmath>
#include<queue>
#include <bitset>
#include<unordered_map>
	#ifndef local
	#define endl '\n'
#endif */
#define mkp make_pair
using namespace std;
using std::bitset;
typedef long long ll;
typedef long double ld;
const int inf=0x3f3f3f3f;
const ll MAXN=2e6+10;
const ll N=1e5+100;
const ll mod=1e9+7;
const ll hash_p1=1610612741;
const ll hash_p2=805306457;
const ll hash_p3=402653189;
//-----------------------------------------------------------------------------------------------------------------*/
// ll head[MAXN],net[MAXN],to[MAXN],edge[MAXN]/*流量*/,cost[MAXN]//费用;
/* 
void add(ll u,ll v,ll w,ll s){
	to[++cnt]=v;net[cnt]=head[u];edge[cnt]=w;cost[cnt]=s;head[u]=cnt;
	to[++cnt]=u;net[cnt]=head[v];edge[cnt]=0;cost[cnt]=-s;head[v]=cnt;
}
struct elemt{
	int p,v;
};
struct comp{
	public:
		bool operator()(elemt v1,elemt v2){
			return v1.v<v2.v;
		}
};
-----------------------------------
求[1,MAXN]组合式和逆元 
ll mi(ll a,ll b){
	ll res=1;
	while(b){
		if(b%2){
			res=res*a%mod;
		}	
		a=a*a%mod;
	}
	return res;
}
ll fac[MAXN],inv[MAXN]
fac[0]=1;inv[0]=1;
for(int i=1;i<=MAXN;i){
	fac[i]=(fac[i-1]*i)%mod;
	inv[i]=mi(fac[i],mod-2);
}
ll C(int m,int n){//组合式C(m,n); 
	if(!n){
		return 1;
	}
	return fac[m]*(inv[n]*inv[m*-n]%mod)%mod;
}
---------------------------------
 unordered_map<int,int>mp;
//优先队列默认小顶堆 ， greater<int> --小顶堆  less<int> --大顶堆  
priority_queue<elemt,vector<elemt>,comp>q;
	set<int>::iterator it=st.begin();
*/
// vector<vector<int>>edge; 二维虚拟储存坐标 
//-----------------------------------------------------------------------------------------------------------------*/
  //map<int,bool>mp[N]; 
int b[10*N][3];//树状数组存
int sum[2*N];//记录当前位置‘-’比‘+’多多少个
int n;
void update(int lei,int x,int v){
	while(x<=n+1){
		b[x][lei]+=v;
		x+=x&-x;
	}
}
ll query(int lei,int x){
	int res=0;
	while(x){
		res+=b[x][lei];
		x-=x&-x;
	}
	return res;
}
int main(){
/*cout<<setiosflags(ios::fixed)<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（不含整数部分）*/
/*cout<<setprecision(8)<<ans<<endl;//输出ans（float）格式控制为8位小数（含整数部分）*/
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);//同步流
	int t;
	cin>>t;
	while(t--){
		cin>>n;
		string s;
		cin>>s;
		for(int i=0;i<=n+10;i++){//清空树状数组
			for(int j=0;j<3;j++){
				b[i][j]=0;
			}
		}
		vector<int>c;
		for(int i=1;i<=n;i++){
			if(s[i-1]=='+'){//注意sum是存‘-’比‘+’多多少个
				sum[i]=sum[i-1]-1;
			}
			else{
				sum[i]=sum[i-1]+1;
			}
			c.push_back(sum[i]);
		}
		c.push_back(-1e9);//因为树状数组下标从1开始，插个极小值使得查询下标从1开始 
		sort(c.begin(),c.end());
		unique(c.begin(),c.end());
		ll ans=0;
		for(int i=0;i<=n;i++){
			int d=lower_bound(c.begin(),c.end(),sum[i])-c.begin();
			ans+=query((sum[i]%3+3)%3,d);//query查询有多少个位置在我前面并且sum比当前小而且二者对3取余的余数相同
			update((sum[i]%3+3)%3,d,1);
		}
		cout<<ans<<endl;
	}
	return 0;
}