转载

第七届蓝桥杯Java B——压缩变换

小明最近在研究压缩算法。

他知道，压缩的时候如果能够使得数值很小，就能通过熵编码得到较高的压缩比。

然而，要使数值很小是一个挑战。

最近，小明需要压缩一些正整数的序列，这些序列的特点是，后面出现的数字很大可能是刚出现过不久的数字。对于这种特殊的序列，小明准备对序列做一个变换来减小数字的值。

变换的过程如下：

从左到右枚举序列，每枚举到一个数字，如果这个数字没有出现过，刚将数字变换成它的相反数，如果数字出现过，则看它在原序列中最后的一次出现后面（且在当前数前面）出现了几种数字，用这个种类数替换原来的数字。

比如，序列(a1, a2, a3, a4, a5)=(1, 2, 2, 1, 2)在变换过程为：

a1: 1未出现过，所以a1变为-1；

a2: 2未出现过，所以a2变为-2；

a3: 2出现过，最后一次为原序列的a2，在a2后、a3前有0种数字，所以a3变为0；

a4: 1出现过，最后一次为原序列的a1，在a1后、a4前有1种数字，所以a4变为1；

a5: 2出现过，最后一次为原序列的a3，在a3后、a5前有1种数字，所以a5变为1。

现在，给出原序列，请问，按这种变换规则变换后的序列是什么。

输入格式：

输入第一行包含一个整数n，表示序列的长度。

第二行包含n个正整数，表示输入序列。

输出格式：

输出一行，包含n个数，表示变换后的序列。

例如，输入：

1 2 2 1 2

程序应该输出：

-1 -2 0 1 1

再例如，输入：

1 1 2 3 2 3 1 2 2 2 3 1

程序应该输出：

-1 0 -2 -3 1 1 2 2 0 0 2 2

数据规模与约定

对于30%的数据，n<=1000；

对于50%的数据，n<=30000；

对于100%的数据，1 <=n<=100000，1<=ai<=10^9

资源约定：

峰值内存消耗（含虚拟机） < 256M

CPU消耗 < 3000ms

这题如果不懂题意就多读两遍，其实很好理解，这里不再赘述，说一下做法

题目中涉及到一个操作是查询区间$[l,r]$内不同数字的个数，说到区间查询，很容易想到线段树，但是这题难点在于如果把数字种数这个信息存到线段树中。考虑一下，加入我们用一个数组$b$来存输入样例中，所有数字最后一次出现的位置。具体含义是，假设输入的序列为$1,2,3,1,2,2$，那么$b$数组中，因为1没出现过，所以$b[0]=1$；2也没出现过，所以$b[1]=1$；3也没出现过，所以$b[2]=1$；又看到1，因为1出现过了，所以首先把$b[0]$改为0，然后把$b[3]$改为1；又看到2，所以把$b[1]$改为0，然后把$b[4]$改为1；最后看到2，所以把$b[4]$改为0，然后把$b[5]$改为1。这样做以后，区间[l,r]内不同数字的个数就是数组$b$的下标从$l$到$r$内所有数字的和，那么其实就可以直接用线段树去维护数组$b$的和

判断一个数出没出现过只需要往Map里面存就行了

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class Main {

    static Map<Integer, Integer> map = new HashMap<Integer, Integer>();
    static int[] arr; // data sequence
    static int[] ans; // result sequence
    static int[] b; // 01 sequence if b[i] == 1 that means arr[p] the last place was p
    static Node[] node; // Segement Tree Node

    public static void main(String[] args) {
        Scanner cin = new Scanner(System.in);
        int n = cin.nextInt();
        node = new Node[n << 2];
        arr = new int[n];
        b = new int[n];
        ans = new int[n];
        build(0, n - 1, 0);
        for (int i = 0; i < n; i++) {
            arr[i] = cin.nextInt();
            Integer pre = map.get(arr[i]);
            if (pre == null) {
                ans[i] = -arr[i];
                b[i] = 1;
                update(i, 1, 0);
            } else {
                // The number of different types of Numbers in a statistical interval
                ans[i] = query(pre + 1, i - 1, 0);
                b[pre] = 0;
                b[i] = 1;
                update(pre, -1, 0);
                update(i, 1, 0);
            }
            map.put(arr[i], i);
        }
        for (int i = 0; i < n; i++)
            System.out.printf("%d ", ans[i]);
    }

    private static void update(int i, int val, int idx) {
        node[idx].sum += val;
        if (node[idx].l == node[idx].r)
            return;
        int mid = (node[idx].l + node[idx].r) >> 1;
        if (i <= mid)
            update(i, val, (idx << 1) | 1);
        else
            update(i, val, (idx << 1) + 2);
    }

    static void build(int l, int r, int idx) {
        node[idx] = new Node(l, r);
        if (l == r) {
            node[idx].sum = b[l];
        } else {
            int mid = (node[idx].l + node[idx].r) >> 1;
            build(l, mid, (idx << 1) | 1);
            build(mid + 1, r, (idx << 1) + 2);
            node[idx].sum = node[(idx << 1) | 1].sum + node[(idx << 1) + 2].sum;
        }
    }

    static int query(int l, int r, int idx) {
        if (l <= node[idx].l && r >= node[idx].r)
            return node[idx].sum;
        int mid = (node[idx].l + node[idx].r) >> 1;
        int ans = 0;
        if (l <= mid) ans += query(l, r, (idx << 1) | 1);
        if (r > mid) ans += query(l, r, (idx << 1) + 2);
        return ans;
    }
}

class Node {
    int l, r, sum;

    Node(int l, int r) {
        this.l = l;
        this.r = r;
    }
}

原文 https://www.wmathor.com/index.php/archives/1267/

正文到此结束