【LEETCODE】73、根据身高重建队列 第406题

说实话，这道题我没想出来，但是看解题报告题解比较让人觉得眼前一亮，这里记录下来

package y2019.Algorithm.greedy.medium;

import java.util.Arrays;
import java.util.LinkedList;

/**
 * @Auther: xiaof
 * @Date: 2019/12/19 09:05
 * @Description: 406. 根据身高重建队列
 *
 * 假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。
 * 注意：
 * 总人数少于1100人。
 *
 * 示例
 * 输入:
 * [[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
 * 输出:
 * [[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/queue-reconstruction-by-height
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 */
public class ReconstructQueue {

    /**
     * https://leetcode-cn.com/problems/queue-reconstruction-by-height/solution/406-gen-ju-shen-gao-zhong-jian-dui-lie-pai-xu-hou-/
     * 执行用时 : 10 ms , 在所有 java 提交中击败了 76.85% 的用户
     * 内存消耗 : 40 MB , 在所有 java 提交中击败了 99.37% 的用户
     */
    public int[][] solution(int[][] people) {
        //1.思路就是先按照升高排序,先按照升高排序，升高相同的按照第二个前面几个人进行排序
        Arrays.sort(people, (v1, v2) -> v1[0] == v2[0] ? v1[1] - v2[1] : v2[0] - v1[0]);
        //2.然后遍历数组，依次吧中的升高添加到数据中，然后把
        //3.根据排好顺序的list，吧第二个参数作为下坐标add进入队列中
        LinkedList<int[]> res = new LinkedList<>();
        //因为已经加入linked的数据都是比当前遍历到的数据大的值，那么只要知道前面有几个比这个数大，直接插入即可
        for (int[] i : people) {
            res.add(i[1], i);
        }

        return res.toArray(new int[res.size()][2]);

    }

    public static void main(String[] args) {
        int[][] param1 = {{7,0}, {4,4}, {7,1}, {5,0}, {6,1}, {5,2}};

        ReconstructQueue fuc = new ReconstructQueue();
        System.out.println("需要添加对应的数据:");
        int[][] res = fuc.solution(param1);
        System.out.println("");
        System.out.println(res);

    }

}

2020年1月19日08:54:11 add  》》》》》》》》》》》》》》》》》

如果使用快排进行排序速度会更快

package y2019.Algorithm.greedy.medium;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

/**
 * @Auther: xiaof
 * @Date: 2019/12/19 09:05
 * @Description: 406. 根据身高重建队列  贪心算法
 *
 * 假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。
 * 注意：
 * 总人数少于1100人。
 *
 * 示例
 * 输入:
 * [[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
 * 输出:
 * [[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/queue-reconstruction-by-height
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 */
public class ReconstructQueue {

    //使用快排

    /**
     * 执行用时 : 5 ms , 在所有 Java 提交中击败了 99.95% 的用户
     * 内存消耗 : 43.1 MB , 在所有 Java 提交中击败了 93.71% 的用户
     * @param people
     * @return
     */
    public int[][] solution2(int[][] people) {
        //先进行快排
        quikSort(people, 0, people.length);
        //按照顺序插入list
//        List<int[]> res = new ArrayList();
        LinkedList<int[]> res = new LinkedList<>();
        for (int[] i : people) {
            res.add(i[1], i);
        }

        return res.toArray(new int[res.size()][]);
    }

    public void quikSort(int[][] people, int start, int end) {
        if (start < end) {
            int mid = midSite(people, start, end);
            quikSort(people, start, mid);
            quikSort(people, mid + 1, end);
        }
    }

    public int midSite(int[][] people, int start, int end) {
        if (start >= end) {
            return start;
        }

        int midValue[] = people[start], l = start, r = end;
        do {
            do {
                ++l;
            } while (l < end && compare(people[l], midValue));

            do {
                --r;
            } while (start < r && compare(midValue, people[r]));

            //如果没有超出范围，交换位置
            if (l < r) {
                int[] tmp = people[l];
                people[l] = people[r];
                people[r] = tmp;
            }
        } while (l < r);

        //最后吧之前中间位置数据交换回来
        people[start] = people[r];
        people[r] = midValue;

        return r;
    }

    //这里我们按照从大到小排序
    public boolean compare(int[] p1, int[] p2) {
        //比较，优先根据第二个数据比较，然后根据第一个数据比较大小
        if (p1[0] != p2[0]) {
            return p1[0] > p2[0];
        } else {
            //如果相等,第一个数据
            return p1[1] < p2[1];
        }
    }


    public static void main(String[] args) {
        int[][] param1 = {{7,0}, {4,4}, {7,1}, {5,0}, {6,1}, {5,2}};

        ReconstructQueue fuc = new ReconstructQueue();
        System.out.println("需要添加对应的数据:");
        int[][] res = fuc.solution2(param1);
        System.out.println("");
        System.out.println(res);

    }
}