第三个样例解释:{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000
模拟。
找到b所在位置为p。
记录1-p的s[i]表示i到p之间比b大的数的个数-比b小的数的个数。
而p到n的s‘[i]则与s[i]相反,并用cnt[k]来记录s‘[i]=k的有几个。
那么左右可以配对需要满足的条件就是s[i]=-s‘[i],那么直接从1-p扫一遍,加上对应的cnt[-s[i]]即可。
对于负数整体加上一个数,让他变成正的即可。
#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> #define k 100000 #define LL long long using namespace std; int n,b,a[k+5],cnt[k*2+5],s[k+5]; void read(int &tmp) { tmp=0; char ch=getchar(); int fu=1; for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') fu=-1; for (;ch>='0'&&ch<='9';ch=getchar()) tmp=tmp*10+ch-'0'; tmp*=fu; } int main() { read(n),read(b); int p=0; for (int i=1;i<=n;i++) { read(a[i]); if (a[i]==b) p=i; } for (int i=p-1;i;i--) if (a[i]>b) s[i]=s[i+1]+1; else s[i]=s[i+1]-1; int now=0; cnt[k]++; for (int i=p+1;i<=n;i++) { if (a[i]>b) now++; else now--; cnt[now+k]++; } LL ans=0; for (int i=1;i<=p;i++) ans+=cnt[k-s[i]]; printf("%lld\n",ans); return 0; }
原文地址:http://blog.csdn.net/regina8023/article/details/44014389