Time Limit: 12000/6000 MS (Java/Others)
Memory Limit: 65536/65536 K (Java/Others)
题目大意:
给出n个数的数列(两两不同),对于每一个数,求出以这个数字为中位数的区间个数有多少。
解题思路:
估计了一下时间复杂度,大致要有一个n^2的算法,加logn也不行。
那么可以把这n个数一个个的分开算,时间复杂度用掉一个n,如何在n的时间内求出一个数是多少个区间的中位数?
注意到一个区间 以某一个数为中位数可以转化为在这个区间中比他小的数个数和比他大的个数相等,所以对每一个数,开两个数组l,r,定义为:他左或者右方的以他为一端端点的区间里,对每个区间,得到一个比他大的数个数-比他小的个数的值,然后存到l,r里(下标是差值,存的是区间个数)答案就是左右区间中这个下标值互为相反数的对应值乘积+这些区间中下标为0的区间数+1(这个1是因为他本身一个数做一个区间也算中位数)
p.s 1.答案可能爆int,应该开成long long
2.如果比他大的数个数-比他小的个数<0,那么没法存成下标,所以应该在这个下标+n,防止向下溢出。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#define INF 0x3f3f3f3f
#define c(a) memset(a,0,sizeof a)
#define C_1(a) memset(a,-1,sizeof a)
#define C_i(a) memset(a,0x3f,sizeof a)
#define rush() int t;scanf("%d",&t);while(t--)
#define ll long long
using namespace std;
int a[16080],ans[16080],l[16080],r[16080];
int main()
{
int n;
while (scanf("%d", &n) == 1)
{
c(ans);
int s = n+10;
for (int i = 0; i < n; i++)scanf("%d", &a[i]);
for (int i = 0; i < n; i++)
{
ans[i]++;
c(l), c(r);
int tem = 0;
int lmin = s,lmax=s;
for (int j = i-1; j>=0 ; j--)
{
if (a[j] < a[i])
{
l[--tem + s]++;
lmin = min(lmin, tem+s);
}
else
{
l[++tem + s]++;
lmax = max(lmax, tem+s);
}
}
tem = 0;
for (int j = i + 1; j < n; j++)
{
if (a[j] < a[i])r[--tem + s]++;
else r[++tem + s]++;
}
ans[i] += l[s] + r[s];
for (int j = lmin; j <= lmax; j++)
ans[i] += l[j] * r[2 * s - j];
}
for (int i = 0; i < n; i++)
{
printf("%d%c", ans[i], i == n - 1 ? '\n' : ' ');
}
}
}