Submit: 3544 Solved: 2194
[Submit][Status][Discuss]
Description
给出1~n的一个排列,统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后,位于中间的数。
Input
第一行为两个正整数n和b ,第二行为1~n 的排列。
Output
输出一个整数,即中位数为b的连续子序列个数。
Sample Input
7 4
5 7 2 4 3 1 6
5 7 2 4 3 1 6
Sample Output
4
HINT
第三个样例解释:{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000
非常机智的思维题
将比b小的数计为-1,比b大的数计为1,记录中位数出现的位置为pos
设l[sum[i]]为i到pos的和为sum[i]出现的次数
r[sum[i]]为pos到i的和为sum[i]出现的次数
ans=l[sum[i]]*r[-sum[i]]
处理一下数组下标
1 #include<iostream> 2 #include<cstdio> 3 using namespace std; 4 5 const int MAXN=200005; 6 int n,b,pos,ans; 7 int a[MAXN],sum[MAXN],l[MAXN],r[MAXN]; 8 9 int main() 10 { 11 scanf("%d%d",&n,&b); 12 for(int i=1;i<=n;i++) 13 { 14 scanf("%d",&a[i]); 15 if(a[i]>b) a[i]=1; 16 else if(a[i]<b) a[i]=-1; 17 else {a[i]=0;pos=i;} 18 } 19 l[n]=r[n]=1; 20 for(int i=pos-1;i>=1;i--) 21 { 22 sum[i]=sum[i+1]+a[i]; 23 l[sum[i]+n]++; 24 } 25 for(int i=pos+1;i<=n;i++) 26 { 27 sum[i]=sum[i-1]+a[i]; 28 r[sum[i]+n]++; 29 } 30 for(int i=0;i<2*n;i++) 31 ans+=l[i]*r[2*n-i]; 32 printf("%d ",ans); 33 return 0; 34 }