题目传送门(内部题37)


输入格式

第一行一个整数$n$,表示区间的长度.
第二行一个长度为$n$的只包含$0,1,2$的字符串,表示给出的序列。


输出格式

一行一个整数,表示革命的区间的数量。


样例

样例输入:

10
0000111011

样例输出:


数据范围与提示

第$1$个测试点,$n=100$
第$2,3$个测试点,$n=1,000$
第$4$个测试点,$n=50,000$
第$5,6$个测试点,$n={10}^5$
第$7,8,9,10$个测试点,$n=5\times {10}^6$
第$2,4,5,7$个测试点还满足:给出的序列中只含$0$和$1$


题解

显然需要用到前缀和,那么我们设$s_0,s_1,s_2$分别表示$0,1,2$的前缀和。

那么对于一个区间$(l,r)$,只有当这个区间满足$s_r-s_{l-1}>\frac{r-l+1}{2}$的时候满足,化简式子,$2\times s_r+r-1>2\times s_{l-1}+l$。

但是显然这样求会重复,那么我么考虑这样一个问题,对于一个区间,$0,1,2$中只有一个可能会超过区间的一半,所以我们可以找出不满足的个数,用总的减去即可。

那么我们所要求的就是当前位置以前有几个点满足这个性质,可以用树状数组快速求出,维护三个树状数组即可。

我的做法稍特殊,首先在每个树状数组开始插一个$0$,然后我们只要求出小于等于$s_i-i$的个数即可。

时间复杂度:$\Theta(n\log n)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
int n;
char ch[5000001];
int a[5000001];
int s[3][5000001];
int tr0[20000010],tr1[20000010],tr2[20000010];
long long ans;
int lowbit(int x){return x&-x;}
void add0(int x){for(int i=x;i<=10000000;i+=lowbit(i))tr0[i]++;}
int ask0(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i))res+=tr0[i];
return res;
}
void add1(int x){for(int i=x;i<=10000000;i+=lowbit(i))tr1[i]++;}
int ask1(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i))res+=tr1[i];
return res;
}
void add2(int x){for(int i=x;i<=10000000;i+=lowbit(i))tr2[i]++;}
int ask2(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i))res+=tr2[i];
return res;
}
int main()
{
scanf("%d%s",&n,ch+1);
for(int i=1;i<=n;i++)
{
a[i]=ch[i]-'0';
s[a[i]][i]++;
s[0][i]+=s[0][i-1];
s[1][i]+=s[1][i-1];
s[2][i]+=s[2][i-1];
}
add0(5000001);
add1(5000001);
add2(5000001);
for(int i=1;i<=n;i++)
{
ans+=i;
ans-=ask0(2*s[0][i]-i+5000000);
ans-=ask1(2*s[1][i]-i+5000000);
ans-=ask2(2*s[2][i]-i+5000000);
add0(2*s[0][i]-i+5000001);
add1(2*s[1][i]-i+5000001);
add2(2*s[2][i]-i+5000001);
}
printf("%lld",ans);
return 0;
}

rp++

[CSP-S模拟测试]:斯诺(snow)(数学+前缀和+树状数组)的更多相关文章

  1. [CSP-S模拟测试]:降雷皇(DP+树状数组)

    题目描述 降雷皇哈蒙很喜欢雷电,他想找到神奇的电光.哈蒙有$n$条导线排成一排,每条导线有一个电阻值,神奇的电光只能从一根导线传到电阻比它大的上面,而且必须从左边向右传导,当然导线不必是连续的.哈蒙想 ...

  2. 【CSP模拟赛】奇怪的队列(树状数组 &二分&贪心)

    题目描述 nodgd的粉丝太多了,每天都会有很多人排队要签名.  今天有n个人排队,每个人的身高都是一个整数,且互不相同.很不巧,nodgd今天去忙别的事情去了,就只好让这些粉丝们明天再来.同时nod ...

  3. 【洛谷】NOIP提高组模拟赛Day2【动态开节点/树状数组】【双头链表模拟】

    U41571 Agent2 题目背景 炎炎夏日还没有过去,Agent们没有一个想出去外面搞事情的.每当ENLIGHTENED总部组织活动时,人人都说有空,结果到了活动日,却一个接着一个咕咕咕了.只有不 ...

  4. BZOJ-2743: [HEOI2012]采花 前缀和 树状数组

    BZOJ-2743 LUOGU:https://www.luogu.org/problemnew/show/P4113 题意: 给一个n长度的序列,m次询问区间,问区间中出现两次及以上的数字的个数.n ...

  5. Gym - 101630G The Great Wall (前缀和+树状数组+二分)

    题意:有一个序列,一开始所有的元素都是ai,你可以选择两个长度相等的区间,如果某个元素被一个区间覆盖,那么变为bi,如果被两个区间都覆盖,那么变为ci.问所有区间的选择方法中产生的第k小的元素总和. ...

  6. HDU 3303 Harmony Forever 前缀和+树状数组||线段树

    Problem Description We believe that every inhabitant of this universe eventually will find a way to ...

  7. 「模拟赛20180307」三元组 exclaim 枚举+树状数组

    题目描述 给定 \(n,k\) ,求有多少个三元组 \((a,b,c)\) 满足 \(1≤a≤b≤c≤n\)且\(a + b^2 ≡ c^3\ (mod\ k)\). 输入 多组数据,第一行数据组数\ ...

  8. [CSP-S模拟测试]:Equation(数学+树状数组)

    题目描述 有一棵$n$个点的以$1$为根的树,以及$n$个整数变量$x_i$.树上$i$的父亲是$f_i$,每条边$(i,f_i)$有一个权值$w_i$,表示一个方程$x_i+x_{f_i}=w_i$ ...

  9. [CSP-S模拟测试]:小P的单调数列(树状数组+DP)

    题目描述 小$P$最近喜欢上了单调数列,他觉得单调的数列具有非常多优美的性质.经过小$P$复杂的数学推导,他计算出了一个单调增数列的艺术价值等于该数列中所有书的总和.并且以这个为基础,小$P$还可以求 ...

随机推荐

  1. [20190727NOIP模拟测试9]单(single) 题解(树上dp)

    啊啊啊啊啊啊啊啊考场上差一点就A掉了5555 千里之堤溃于蚁穴……鬼知道最后一步那么显然的柿子我为什么没考虑用上…… 观察数据范围可知,出题人期望我们想出一个$O(n)$的做法 当然也有可能是$O(n ...

  2. CPUID

    http://en.wikipedia.org/wiki/CPUID CPUID From Wikipedia, the free encyclopedia    The CPUID opcode i ...

  3. 项目搭建(三):自定义DLL

    说明:程序中有些自定义的控件类型在TestStack.White框架中没有涉及,需要引入自定义的DLL,通过鼠标点击事件处理 使用:将自定义的ClassLibrary2.dll拷贝到项目/bin/de ...

  4. Could not resolve placeholder'XXX' in string value "XXXX"

    练习SSM项目的demo中遇到一个问题,我在applicationContext.xml中使用了<context:property-placeholder location="clas ...

  5. 斯坦福【概率与统计】课程笔记(二):从EDA开始

    探索性数据分析(Exploratory Data Analysis) 本节课程先从统计分析四步骤中的第二步:EDA开始. 课程定义了若干个术语,如果学习过机器学习的同学,应该很容易类比理解: popu ...

  6. golang 国内环境配置

    写在 .bashrc 或 .zshrc 等文件中: # golang env export GO111MODULE=on export GOPROXY=https://goproxy.cn #七牛云提 ...

  7. python使用消息队列RabbitMq(入门)

    windows平台开发和使用 安装 安装Erlang:https://pan.baidu.com/s/1QcZDaI205uaue7mMWh5cSA 安装RabbitMQ:https://pan.ba ...

  8. box-shadow 制作单边阴影效果,不影响其它边的效果

    box-shadow 制作单边阴影效果,不影响其它边的效果:  https://blog.csdn.net/u010289111/article/details/53171128 CSS 样式实现单边 ...

  9. Android No static field XXX of type I in class Lcom/XXX/R$id错

    问题复现: 问题原因: 出现这样的情况,你先检查你的依赖工程(module)的对应布局layout/xxx.xml是否跟主项目的layout重名,你点开R文件的时候,你会发现你的布局发生了错乱,导致你 ...

  10. sql datetime类型数据如果进行模糊查询

    select * from Table1 where CONVERT(nvarchar(50),CreateTime,120) like '%2019'