BestCoder Sequence
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 1616 Accepted Submission(s): 567
Problem Description
Mr Potato is a coder.
Mr Potato is the BestCoder.
One night, an amazing sequence appeared in his dream. Length of this sequence is odd, the median number is M, and he named this sequence as Bestcoder Sequence.
As the best coder, Mr potato has strong curiosity, he wonder the number of consecutive sub-sequences which are bestcoder sequences in a given permutation of 1 ~ N.
Mr Potato is the BestCoder.
One night, an amazing sequence appeared in his dream. Length of this sequence is odd, the median number is M, and he named this sequence as Bestcoder Sequence.
As the best coder, Mr potato has strong curiosity, he wonder the number of consecutive sub-sequences which are bestcoder sequences in a given permutation of 1 ~ N.
Input
Input contains multiple test cases.
For each test case, there is a pair of integers N and M in the first line, and an permutation of 1 ~ N in the second line.
[Technical Specification]
1. 1 <= N <= 40000
2. 1 <= M <= N
For each test case, there is a pair of integers N and M in the first line, and an permutation of 1 ~ N in the second line.
[Technical Specification]
1. 1 <= N <= 40000
2. 1 <= M <= N
Output
For each case, you should output the number of consecutive sub-sequences which are the
Bestcoder Sequences.
Sample Input
1 1 1 5 3 4 5 3 2 1
Sample Output
1 3HintFor the second case, {3},{5,3,2},{4,5,3,2,1} are Bestcoder Sequence.
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 50000
int a[N],dp[N];
int sum[N*3];
int main()
{
int n,m;
while(~scanf("%d %d",&n,&m))
{
int v;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]==m) v=i;
}
int ans=1;
memset(sum,0,sizeof(sum));
dp[v]=0;
for(int i=v-1;i>=1;i--)
{
if(a[i]>m) dp[i]=dp[i+1]+1;
else dp[i]=dp[i+1]-1;
if(!dp[i]) ans++;
sum[dp[i]+N]++;
}
for(int i=v+1;i<=n;i++)
{
if(a[i]>m) dp[i]=dp[i-1]+1;
else dp[i]=dp[i-1]-1;
if(!dp[i]) ans++;
if(sum[N-dp[i]]) ans+=sum[N-dp[i]];
}
printf("%d\n",ans);
}
return 0;
}