POJ 2352 Stars 线段树 数星星
题意:已知n个星星的坐标。每个星星都有一个等级,数值等于坐标系内纵坐标和横坐标皆不大于它的星星的个数。星星的坐标按照纵坐标从小到大的顺序给出,纵坐标相同时则按照横坐标从小到大输出。 (0 <= x, y <= 32000) 要求输出等级0到n-1之间各等级的星星个数。详细信息点击进入传送门。
思路:这个题目数据的输出顺序给了很大的便利。由于是按照纵坐标从小到大输入,因此先输入的星星一定在后来的星星的下面。 根据等级的计算规则,每次输入一颗星星的坐标,只要计算出横坐标不超过它的星星个数。 在x轴上建线段树,范围从0到32000,统计横坐标出现在区间内的星星个数。假设当前输入的星星坐标为x, y, 首先查询[0, x]区间内的星星个数,即为它的等级; 然后将该星星插入线段树。
1 #include<stdio.h> 2 #include<string.h> 3 #define maxn 33333 4 #define lson l, m, rt << 1 5 #define rson m + 1, r, rt << 1 | 1 6 int sum[maxn<<2], rank[maxn]; 7 int query(int s,int l,int r,int rt) 8 { 9 if (r <= s) return sum[rt]; 10 else if (s < l) return 0; 11 int m = (l + r) >> 1; 12 int ret = 0; 13 if (s <= m) ret += query(s, lson); 14 else 15 { 16 ret += query(s, lson); 17 ret += query(s, rson); 18 } 19 return ret; 20 } 21 void PushUp(int rt) 22 { 23 sum[rt] = sum[rt<<1] + sum[rt<<1|1]; 24 } 25 void update(int s,int l,int r,int rt) 26 { 27 if (l == r && l == s) 28 { 29 sum[rt] ++; 30 return; 31 } 32 int m = (l + r) >> 1; 33 if (s <= m) update(s, lson); 34 else update(s, rson); 35 PushUp(rt); 36 } 37 int main() 38 { 39 int n; 40 //freopen("data.in","r",stdin); 41 scanf("%d",&n); 42 memset(sum, 0, sizeof(sum)); 43 memset(rank, 0, sizeof(rank)); 44 for (int i = 0; i < n; i++) 45 { 46 int x, y; 47 scanf("%d%d",&x,&y); 48 rank[query(x, 0, 32000, 1)]++; 49 update(x, 0, 32000, 1); 50 } 51 for (int i = 0; i < n; i++) 52 printf("%d\n",rank[i]); 53 return 0; 54 }