POJ 2352 Stars(线段树单点更新)

来源:互联网 发布:中大网络继续教育学院 编辑:程序博客网 时间:2024/06/05 02:41

题解转自该博客
http://blog.csdn.net/shuangde800/article/details/8175139

题意:

在坐标上有n个星星,如果某个星星坐标为(x, y), 它的左下位置为:(x0,y0),x0<=x 且y0<=y。如果左下位置有a个星星,就表示这个星星属于level x
按照y递增,如果y相同则x递增的顺序给出n个星星,求出所有level水平的数量。

解析:

因为输入是按照按照y递增,如果y相同则x递增的顺序给出的,所以,对于第i颗星星,它的level就是之前出现过的星星中,横坐标x小于等于i星横坐标的那些星星的总数量(前面的y一定比后面的y小)。
所以,需要找到一种数据结构来记录所有星星的x值,方便的求出所有值为0~x的星星总数量。
树状数组和线段树都是很适合处理这种问题。

总结:

边界应该是[1, max]但是我错写成[1, n],导致debug了好久。

AC代码

#include <cstring>#include <cstdio>#include <algorithm>#define ls o*2#define rs o*2+1using namespace std;const int N = 32000 + 10;struct Star {    int x, y;}star[N];int sumv[N << 2], ans[N];bool cmp(Star a, Star b) {    if(a.y != b.y)        return a.y < b.y;    return a.x < b.x;}int ql, qr;int query(int o, int L, int R) {    if(ql <= L && R <= qr) return sumv[o];    int M = (L + R)/2, ret = 0;    if(ql <= M) ret += query(ls, L, M);    if(M < qr) ret += query(rs, M+1, R);    return ret;}int p;void modify(int o, int L, int R) {    if(L == R) {        sumv[o]++;        return ;    }    int M = (L + R)/2;    if(p <= M) modify(ls, L, M);    else modify(rs, M+1, R);    sumv[o] = sumv[ls] + sumv[rs];}int main() {    //freopen("in.txt", "r", stdin);    int n;    while(~scanf("%d", &n)) {        memset(sumv, 0, sizeof(sumv));        memset(ans, 0, sizeof(ans));        int maxn = 0;        for(int i = 1; i <= n; i++) {            scanf("%d%d", &star[i].x , &star[i].y);            star[i].x++;            maxn = max(maxn, star[i].x);        }        sort(star+1, star+1+n, cmp);        for(int i = 1; i <= n; i++) {            ql = 1, qr = star[i].x;            ans[query(1, 1, maxn)]++;            p = star[i].x;            modify(1, 1, maxn);        }        for(int i = 0; i < n; i++) {            printf("%d\n", ans[i]);        }    }    return 0;}
0 0
原创粉丝点击