POJ 2481 Cows (线段树)

题目类型  线段树

题目意思

给出最多100000头牛的两个属性S和E 对于牛i 和牛j 如果 Si <= Sj && Ei >= Ej && Ei - Si > Ej - Sj 则牛i比牛j强壮

现在问对于所有的牛 比它强壮的牛有多少头

解题方法

从题目的条件可以发现 如果 Ei >= Ej && Si <= Sj 满足的话 Ei - Si >= Ej - Sj 也会成立 所以就两个值都相同的情况特殊一点而已

那么可以先把所有牛按E从大到小排序如果E相同就按S从小到大排序

这样如果对于排好序后的数组从左到右处理 假设目前处理到第 i 头牛 前 i-1 头牛的S值已经插入到一棵线段树中 那么已经插入线段树中的S对应的E是满足

E >= Ei的(因为是按E从大到小排序的)　所以这时候询问一下 [0, Si] 这个区间的牛的数量就是结果

注意两头牛两个属性值都相同的情况 在从左到右处理的时候可以记录一下对于目前正在处理的牛前面有多少头牛两个属性值都与它相同的 前面询问到的值减去这个值即为有多少头牛是比牛i强壮

参考代码 - 有疑问的地方在下方留言 看到会尽快回复的

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

#define ls (rt<<1)
#define rs ((rt<<1)|1)
#define mid ((l+r)>>1)

const int maxn = 1e5 + 10;

struct P {
	int s, e, pos;
	bool operator < (const P & rhs) const {
		if(e == rhs.e) return s < rhs.s;
		else return e > rhs.e;
	}
}A[maxn];

int ans[maxn];
int num[maxn*4];

void insert(int rt, int l, int r, int x) {
	if(l == r && l == x) { num[rt]++; return ; }
	if(x <= mid) insert(ls, l, mid, x);
	else insert(rs, mid + 1, r, x);
	num[rt] = num[ls] + num[rs];
}

int query(int rt, int l, int r, int L, int R) {
	if(l == L && r == R) return num[rt];
	if(R <= mid) return query(ls, l, mid, L, R);
	else if(L > mid) return query(rs, mid + 1, r, L, R);
	else return query(ls, l, mid, L, mid) + query(rs, mid + 1, r, mid + 1, R);
}

int main() {
//	freopen("in", "r", stdin);
	int n;
	while(scanf("%d", &n), n) {
		for( int i=0; i<n; i++ ) {
			scanf("%d%d", &A[i].s, &A[i].e);
			A[i].pos = i;
		}
		sort(A, A+n);
		ans[A[0].pos] = 0;
		memset(num, 0, sizeof(num));
		insert(1, 0, 1e5, A[0].s);
		int same = 0;
		for( int i=1; i<n; i++ ) {
			if(A[i].e == A[i-1].e && A[i].s == A[i-1].s) same++;
			else same = 0;
			ans[A[i].pos] = query(1, 0, 1e5, 0, A[i].s) - same;
			insert(1, 0, 1e5, A[i].s);
		}
		for( int i=0; i<n-1; i++ ) printf("%d ", ans[i]);
		printf("%d\n", ans[n-1]);
	}
	return 0;
}