数星星问题

 一 问题描述

星星由平面上的点表示，星星的等级为纵横坐标均不超过自己星星数量（不包括自己）。下图中，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;
    }
}

八 测试