Codeforces Round #780 (Div. 3) F2(树状数组+离散+二分 2100)
Link
概述: 细节蛮多的一道题,公式 F 1 F1 F1已经推出来,区别就是从 O ( n 2 ) O(n^2) O(n2)能够实现变成了 O ( n l o g n ) O(nlogn) O(nlogn)才能实现,那么大致思路就是 O ( n ) O(n) O(n)扫描+每次的 O ( l o g n ) O(logn) O(logn)的操作。用 3 3 3颗树状数组来维护 t r [ 3 ] [ N ] tr[3][N] tr[3][N]。然后因为存在负数所以我们要离散化,但是离散化之后 v e c vec vec的下标是从 0 0 0开始的显然我们要将其变成 1 1 1开始才行,所以我们离散化后调用 f i n d find find函数
int find(int x){
return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;
}
而且因为这一操作,我们多了很多细节问题,例如每一次的树的清空必须清空到 n + 1 n+1 n+1,因为最坏情况没有重复的数,那么我们会用到 t r [ ? ] [ n + 1 ] tr[?][n+1] tr[?][n+1]这个位置。并且由于 3 3 3个 − - −也是一种方案,所以一开始跑的时候需要从 0 0 0开始,我们需要 a d d ( f i n d ( 0 ) , 1 ) add(find(0),1) add(find(0),1)。
- 参考代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
#define int long long
int ss[N];
int tr[5][N];
int mo;
int n;
vector<int>vec;
int find(int x){
return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;//下标从1开始,因为要树状数组
}
int lowbit(int x){
return x&-x;
}
int get(int x){
int res=0;
for(int i=x;i;i-=lowbit(i))res+=tr[mo][i];
return res;
}
void add(int x,int c){
for(int i=x;i<=n+1;i+=lowbit(i))tr[mo][i]+=c;
}
void solve(){
cin>>n;
string s;
cin>>s;
s=" "+s;
for(int i=0;i<=2;i++)//清空树
for(int j=0;j<=n+10;j++)tr[i][j]=0;
vec.clear();
for(int i=1;i<=n;i++){//前缀和
ss[i]=ss[i-1]+(s[i]=='+'?-1:1);
vec.push_back(ss[i]);
}
sort(vec.begin(),vec.end());
vec.erase(unique(vec.begin(),vec.end()),vec.end());//离散
int ans=0;
for(int i=0;i<=n;i++){
mo=(ss[i]%3+3)%3;//可能有负数
int id=find(ss[i]);
ans+=get(id);
add(id,1);
}
cout<<ans<<endl;
}
signed main(){
int t;
cin>>t;
while(t--){
solve();
}
return 0;
}