【洛谷】P4715淘汰赛（线段树的构建）

题目描述

有 （n≤7）个国家参加世界杯决赛圈且进入淘汰赛环节。已经知道各个国家的能力值，且都不相等。能力值高的国家和能力值低的国家踢比赛时高者获胜。1 号国家和 2 号国家踢一场比赛，胜者晋级。3 号国家和 4 号国家也踢一场，胜者晋级……晋级后的国家用相同的方法继续完成赛程，直到决出冠军。给出各个国家的能力值，请问亚军是哪个国家？

输入格式

第一行一个整数 n，表示一共  个国家参赛。

第二行 个整数，第i个整数表示编号为 i的国家的能力值（1≤i≤）。

数据保证不存在平局。

输出格式

仅一个整数，表示亚军国家的编号。

输入输出样例

输入 #1

3

4 2 3 1 10 5 9 7

 输出 #1

1

 解题思路

           本题的关键在于线段树的构建，线段树本质上是一颗二叉树，其中每个节点代表一个区间，对于给定的区间[L , R],一般情况下如果父节点代表区间[L , R]，那么它的两个子节点分别代表区间[L , (L+R)/2]和[(L+R)/2 +1 , R],叶子节点代表区间中的单个元素。

本题中的例#1,对于8个国家，要确定谁是亚军。我们需要将问题一步步拆解：

如上图，要确定8个国家谁是亚军，首先将1~8的国家分为1~4和5~8的国家，然后1~4的国家和5~8的国家再分，直到分为单个国家为止，对于单个国家自己就是冠军，对于两个国家谁的能力大谁就是冠军，依次再向上推，直到确定出最终的冠军。图中蓝色部分即为，线段树的所有节点，满足如果父节点代表区间[L , R]，那么它的两个子节点分别代表区间[L , (L+R)/2]和[(L+R)/2 +1 , R]的条件。画出最终的线段树如下：

可以看到,这颗树也是一颗满二叉树，对于满二叉树,若父节点为n则左右子节点分别为2n和2n + 1,则亚军为tree[2] > tree[3] ? tree[3] : tree[2], 从图中可得亚军为“4”对应的代表区间为1~1，故最终输出结果为1。

源代码：

import java.util.Scanner;
class Tree {
    int power, id;

    Tree(int power, int id) {
        this.power = power;
        this.id = id;
    }
}
public class Main {
    static Tree[] a = new Tree[150];
    static Tree[] tree = new Tree[600];

    static Tree maxt(Tree a, Tree b) {
        return a.power > b.power ? a : b;
    }

    static Tree mint(Tree a, Tree b) {
        return a.power < b.power ? a : b;
    }
    static void build(int node, int start, int end) {
        if (start == end) {
            tree[node] = a[start];
            return;
        }
        int l = node * 2;
        int r = node * 2 + 1;
        int mid = (start + end) / 2;
        build(l, start, mid);
        build(r, mid + 1, end);
        tree[node] = maxt(tree[l], tree[r]);
    }
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n;  // 输入
        n = scanner.nextInt();
        for (int i = 1; i <= (1 << n); i++) {
            int power = scanner.nextInt();
            a[i] = new Tree(power, i);
        }
        build(1, 1, (1 << n));
        System.out.println(mint(tree[2], tree[3]).id);
    }
}

扩展：

本题要求的是亚军，即要求生成线段树时取区间中的最大值，最终可得到结果，只要掌握线段树的生成，也可以根据区间中的最小值，以及区间中所有值的和来创建树，以应对不同题目。