题意:
给定n和m,对于一个由1,2,3...,n组成的序列,问符合下列条件的子序列的个数:
1)子序列连续
2)子序列的中位数为m,
3)子序列长度为奇数
题解:
我们先找到m的位置,然后定义k=0,向左拓展,当遇到的数小于m时k++,否则k--,用f[0][i]存储各种k的出现次数(为了保证非负i=k+maxn);同理向右拓展,当遇到的数小于m时k--,否则k++,用f[1][i]存储k出现的次数。那么结果就是ans=∑(f[0][i]*f[1][i])(maxn-n<=i<=maxn+n)
代码:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <stack>
using namespace std;
const int maxn=4e4+10;
int a[maxn],c[maxn],f[2][maxn*2];
int main()
{
//freopen("D:\\in.txt","r",stdin);
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(f,0,sizeof(f));
int i,j,k=0,flag=0,x;
for(i=0;i<n;i++)
{
scanf("%d",&a[i]);
if(a[i]==m)x=i;
}
k=0;
i=x+1;
f[1][maxn]=f[0][maxn]=1;
for(;i<n;i++)
{
if(a[i]>m)k++;
else k--;
f[0][k+maxn]++;
}
i=x-1;
k=0;
for(;i>=0;i--)
{
if(a[i]>m)k--;
else k++;
f[1][k+maxn]++;
}
int ans=0;
for(i=-n;i<=n;i++)
{
ans+=f[0][i+maxn]*f[1][i+maxn];
}
printf("%d\n",ans);
}
return 0;
}