Codeforces Round #780 (Div. 3) F2(树状数组+离散+二分 2100)

最新推荐文章于 2022-04-10 21:51:45 发布

Snow_raw

最新推荐文章于 2022-04-10 21:51:45 发布

阅读量394

点赞数 1

分类专栏：数据结构文章标签： c++ 算法

本文链接：https://blog.csdn.net/qq_53504307/article/details/123957490

版权

数据结构专栏收录该内容

14 篇文章 0 订阅

订阅专栏

Codeforces Round #780 (Div. 3) F2(树状数组+离散+二分 2100)

Link

概述： 细节蛮多的一道题，公式 $F 1$ 已经推出来，区别就是从 $O(n^2)$ 能够实现变成了 $O (n l o g n)$ 才能实现，那么大致思路就是 $O (n)$ 扫描+每次的 $O (l o g n)$ 的操作。用 $3$ 颗树状数组来维护 $t r [3] [N]$ 。然后因为存在负数所以我们要离散化，但是离散化之后 $v e c$ 的下标是从 $0$ 开始的显然我们要将其变成 $1$ 开始才行，所以我们离散化后调用 $f i n d$ 函数

int find(int x){
	return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;
}

而且因为这一操作，我们多了很多细节问题，例如每一次的树的清空必须清空到 $n + 1$ ，因为最坏情况没有重复的数，那么我们会用到 $t r [?] [n + 1]$ 这个位置。并且由于 $3$ 个 $-$ 也是一种方案，所以一开始跑的时候需要从 $0$ 开始，我们需要 $a d d (f i n d (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;
}