线段树基础 poj2352

题目链接

题目描述

有N个坐标，有一个点o坐标为（x，y） 如果有K个点坐标小于等于它的坐标（即：x0<=x && y0<=y）,那么我们称点O在第K级，即level[k]。、
现要求求出level[0 ~ N-1].
1. N < 15000， x,y < 32000， 无重复点
2. 输入的点Y按照从小到大排序，如果y相同，x从小到大排序

思路

1. 因为输入时已经排好序，所以考虑当前时只需要统计此时x0<=x点的个数
2. 利用线段树O(log(n))查找，因为x,y < 32000，所以可以暴力遍历每个x进行统计

代码如下

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

using namespace std;

const int MAXN = 32010;
int bit[MAXN],level[MAXN];

int sum(int i)
{
    int s = 0;
    while(i>0)
    {
        s += bit[i];
        i -= i&-i;
    }
    return s;
}

void add(int i)
{
    while(i<=MAXN)
    {
        bit[i]++;
        i += i&-i;
    }
}

int main()
{
  int n;
  scanf("%d",&n);
  memset(bit,0,sizeof(bit));
  memset(level,0,sizeof(level));
  for(int i = 0; i<n; i++)
  {
      int x,y;
      scanf("%d%d",&x,&y);
      level[sum(++x)]++; //防止0出现，TLE。因为题意中点明不可能有重合的两个点，所以可以++x（这里还是有问题）。
      add(x);
  }
  for(int i = 0; i<n; i++)
  {
      printf("%d\n",level[i]);
  }
}

如下case：