最小区间问题

作者：Grey

原文地址：最小区间问题

题目描述

LeetCode 632. 最小区间

思路

准备一个数据结构

    public static class Node {
        public int value;// 值是多少
        public int position;// 在链表的哪个位置上
        public int bucket; // 在哪个链表上

        public Node(int v, int p, int b) {
            value = v;
            position = p;
            bucket = b;
        }
    }

对链表中的每个元素都可以做这个数据结构的封装，其中value表示这个节点的值，position表示这个节点在链表中的哪个位置，bucket表示这个节点在哪个链表上。

准备一个有序表，有序表根据值来组织，值小的排在前面，如果值一样，按照链表顺序排；

首先将每个链表头节点都加入这个有序表，然后从有序表中获取第一个元素（最小元素）和最后一个元素（最大元素），这两个元素的区间，一定覆盖所有链表中的至少一个位置的数；

接下来看是否可以收窄区间，每次将有序表的头节点弹出，然后看这个弹出节点所在的链表的下一个位置是否有值，如果有，加入有序表，然后弹出有序表的头位置数据和尾位置数据，看下这个区间是否变的更窄，如此往复。直到某次弹出的值所在的链表已经无数据可加。

完整代码如下：

    public static class Node {
        public int value;// 值是多少
        public int position;// 在链表的哪个位置上
        public int bucket; // 在哪个链表上

        public Node(int v, int p, int b) {
            value = v;
            position = p;
            bucket = b;
        }
    }


    public static int[] smallestRange(List<List<Integer>> nums) {
        if (nums == null) {
            return null;
        }
        if (nums.size() == 1) {
            if (nums.get(0).size() > 0) {
                return new int[]{nums.get(0).get(0), nums.get(0).get(0)};
            } else {
                return null;
            }
        }
        TreeSet<Node> set = new TreeSet<>((o1, o2) -> o1.value != o2.value ? o1.value - o2.value : o1.bucket - o2.bucket);
        int i = 0;
        for (List<Integer> list : nums) {
            set.add(new Node(list.get(0), 0, i));
            i++;
        }
        Node min = set.pollFirst();
        Node max = set.last();
        int[] result = {min.value, max.value};
        while (min.position + 1 < nums.get(min.bucket).size()) {
            set.add(new Node(nums.get(min.bucket).get(min.position + 1), min.position + 1, min.bucket));
            min = set.pollFirst();
            max = set.last();
            result = minRange(result, new int[]{min.value, max.value});
        }
        return result;
    }

    public static int[] minRange(int[] a, int[] b) {
        if (a[1] - a[0] > b[1] - b[0]) {
            return b;
        } else if (a[1] - a[0] < b[1] - b[0]) {
            return a;
        }
        return a[0] > b[0] ? b : a;
    }

更多

算法和数据结构笔记

标签：链表,min,int,bucket,最小,value,问题,区间,public
来源： https://www.cnblogs.com/greyzeng/p/16463385.html