传送门
模拟
- 首先,这道题要求奇数长度,很容易想到可以分类讨论
1.左右两边均为偶数。
2.左右两边都是奇数。
然后从中位数开始左右两边做个前缀和每次左右两边暴力匹配,显然这个是O(n2) - 那么我们考虑优化一下,既然我们不关心具体数值,那么我们就可以用0,-1, 1表示大小关系,那么当左右两边的前缀和相加为0即是一种合法情况。我们可以忽然发现:当左右两边前缀和相加为0,那么这段连续子序列定为奇数
当1 和 -1相匹配时才能等于0 - 现在的复杂度仍然为O(n2),那我们再想一下这个方法的流程,当右边前缀和为k时,我们需要在左边的前缀和中找到值为 -k的与之对应,那么我们需要O(n) 的在左边查找,同时计数,那么我们可以开一个桶
我用的是map ,即可O(1) 完成啦~
初始化
手残选手
代码如下
#include<bits/stdc++.h>
using namespace std;
int n,b,Bit;
int f[100010],add[100010];
map<int,int > dic;
inline int read()
{
int Num=0,f=1; char ch=getchar();
while(ch<'0'||ch>'9') {if(ch=='-') f=-1; ch=getchar();}
while(ch>='0'&&ch<='9') {Num=(Num<<1)+(Num<<3)+ch-'0'; ch=getchar();}
return Num*f;
}
int main()
{
n=read(),b=read();
for(int i=1;i<=n;i++)
{
int x=read();
if(x==b) f[i]=0,Bit=i;
if(x<b) f[i]=-1;
if(x>b) f[i]=1;
}
int x=0;
for(int i=Bit;i>=1;i--) x+=f[i],dic[x]++;
x=0;int ans=dic[0];
for(int i=Bit+1;i<=n;i++) x+=f[i],ans+=dic[-x];
printf("%d",ans);
return 0;
}