51nod-1682 中位数计数

原题链接

1682 中位数计数

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

中位数定义为所有值从小到大排序后排在正中间的那个数，如果值有偶数个，通常取最中间的两个数值的平均数作为中位数。

现在有n个数，每个数都是独一无二的，求出每个数在多少个包含其的区间中是中位数。

Input

第一行一个数n(n<=8000)
第二行n个数，0<=每个数<=10^9

Output

N个数，依次表示第i个数在多少包含其的区间中是中位数。

Input示例

5
1 2 3 4 5

Output示例

1 2 3 2 1

把输入的数存在num数组中,对于num数组的每个元素num[i], 设maxs 为大于num[j]的元素个数，mins为小于num[j]元素个数，遍历j = i-1 to 0, 若num[j] > num[i], maxs++, else mins++, d[8000+maxs - mins]++;遍历完后ans += d[8000]表示0到i中满足条件的区间，maxs = 0, mins = 0, 再遍历i+1到n-1， 若num[j] > num[i] maxs++, else mins++, ans += d[8000+mins-maxs]

ans为num[i]满足条件的区间个数

#include <cstdio>
#include <cstring>
#define maxn 8005
#define MOD 1000000007
using namespace std;
typedef long long ll;

int num[maxn], p[maxn<<1];
int kk[maxn];
int main(){
	
//	freopen("in.txt", "r", stdin);
	int n;
	
	scanf("%d", &n);
	for(int i = 0; i < n; i++)
	 scanf("%d", num+i);
	int e = 0;
	for(int i = 0; i < n; i++){
		memset(p, 0, sizeof(p));
		p[8000] = 1;
		int cnt = 0, ans = 0;
		for(int j = i-1; j >= 0; j--){
			if(num[j] > num[i])
			 cnt++;
			else
			 cnt--;
		    p[cnt+8000]++;
	    }
	    ans += p[8000];
	    cnt = 0;
	    for(int j = i+1; j < n; j++){
	    	if(num[j] > num[i])
	    	 cnt++;
	    	else
	    	 cnt--;
	    	ans += p[-cnt+8000];
	    }
	    kk[e++] = ans;
	}
	for(int i = 0; i < e; i++){
		if(i)
		 putchar(' ');
		printf("%d", kk[i]);
	}
	puts("");
	return 0;
}