一 问题描述
星星由平面上的点表示,星星的等级为纵横坐标均不超过自己星星数量(不包括自己)。下图中,5号星的等级为3。纵横坐标均不超过5号星的星星有3颗,1、2和4号。2、4号星的级别是1。在该地图上有1颗0级星,2颗1级星,1颗2级星。和1颗3级星。计算给定地图上的每个级别的星星数量。
二 输入
第1行包含星星的数量 N,以下 N 行描述星星的坐标,每行都包含两个整数X、Y。平面上的一个点,只可以有一颗星星。以 Y 坐标升序输入,在 Y 坐标相等时以 X 坐标升数输入。
三 输出
输出包含 N 行,第 1 行包含 0 级星星的数量,第 2 行包含 1 级星星的数量,最后 1 行包含 n-1 级星星的数量。
四 输入和输出样例
1 输入样例
5
1 1
5 1
7 1
3 3
5 5
2 输出样例
1
2
1
1
0
五 分析
每个星星的等级都为它左下边星星的个数,输入所有星星的坐标。依次输出等级 0 到 n-1 的星星的数量。1号星星左下没有星星,等级为0,,2号星的左边有1颗星,等级为1,,3号星的左边有两颗星,等级为2,,4号星,左边有一颗星,等级为1,,5号星,左边有3颗星,等级为3。因此等级为0的有1个,等级为1的有2个,等级为2的有1个,等级为3的有1个,等级为4的有0个。
本问题看似二维数据,实际上输入数据已经按照 y 升序,也就是说读到一个点时,当前点的 y 坐标肯定大于或等于已经输入的 y 坐标,如果 y 坐标相等,则 x 坐标肯定大于已经输入的 x 坐标,所以每次只要计算 x 坐标比当前点小的点就行。本问题本质是统计 x 坐标前面星星的数量。是前缀和问题,因为数量较大,暴力穷举会超时,所以可借助树状数组解决。注意,给的点坐标从0 开始。树状数组下标从1开始,0的位置不可用,所以需要在输入 x 坐标时加 1 处理。
六 算法设计
1 依次输入每一个坐标 x、y,执行 x++。
2 计算 x 前缀和 sum(x) ,将其作为该星星的等级,用 ans[] 数组累计该等级的数量。
3 将树状数组中的 x 的数量加1。
七 代码
package com.platform.modules.alg.alglib.poj2352;
public class Poj2352 {
public String output = "";
private int maxn = 32010;
// 等级统计
private int ans[] = new int[maxn];
// 每个值的数量
private int c[] = new int[maxn];
int n;
// c[i]的区间长度
int lowbit(int x) {
return (-x) & x;
}
// 将坐标为 x 的元素增加 val,其后继也要增加
void add(int x, int val) {
while (x <= maxn) {
c[x] += val;
x += lowbit(x);// x 的后继(父结点)
}
}
// 前缀和
int sum(int x) {
int s = 0;
while (x > 0) {
s += c[x];
x -= lowbit(x); // x 的前驱
}
return s;
}
public String cal(String input) {
String[] line = input.split("\n");
n = Integer.parseInt(line[0]);
int x, y;
for (int i = 0; i < n; i++) {
String[] num = line[i + 1].split(" ");
x = Integer.parseInt(num[0]);
y = Integer.parseInt(num[1]);
x++;
ans[sum(x)]++;
add(x, 1); // x的数量 c[x] 增1
}
for (int i = 0; i < n; i++)
output += ans[i] + "\n";
return output;
}
}