排座椅-算法

网上只查到c++代码，补充java代码。

题目链接：434. 排座椅 - AcWing题库

上课的时候总有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。

不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的 DD 对同学上课时会交头接耳。

同学们在教室中坐成了 MM 行 NN 列，坐在第 ii 行第 jj 列的同学的位置是 (i,j)(i,j)，为了方便同学们进出，在教室中设置了 KK 条横向的通道，LL 条纵向的通道。

于是，聪明的小雪想到了一个办法，或许可以减少上课时学生交头接耳的问题：

她打算重新摆放桌椅，改变同学们桌椅间通道的位置，因为如果一条通道隔开了两个会交头接耳的同学，那么他们就不会交头接耳了。

请你帮忙给小雪编写一个程序，给出最好的通道划分方案。

在该方案下，上课时交头接耳的学生对数最少。

输入格式

输入文件的第一行，有 55 个用空格隔开的整数，分别是 M，N，K，L，DM，N，K，L，D。

接下来 DD 行，每行有 44 个用空格隔开的整数，第 ii 行的 44 个整数 Xi，Yi，Pi，QiXi，Yi，Pi，Qi，表示坐在位置 (Xi,YiXi,Yi) 与 (Pi,QiPi,Qi) 的两个同学会交头接耳（输入保证他们前后相邻或者左右相邻）。

输入数据保证最优方案的唯一性。

输出格式

输出文件共两行。

第一行包含 KK 个整数，a1,a2,…,aKa1,a2,…,aK，表示第 a1a1 行和 a1+1a1+1 行之间、第 a2a2 行和第 a2+1a2+1 行之间、…、第 aKaK 行和第 aK+1aK+1 行之间要开辟通道，其中 ai<ai+1ai<ai+1，每两个整数之间用空格隔开（行尾没有空格）。

第二行包含 LL 个整数，b1,b2,…,bLb1,b2,…,bL，表示第 b1b1 列和 b1+1b1+1 列之间、第 b2b2 列和第 b2+1b2+1 列之间、…、第 bLbL 列和第 bL+1bL+1 列之间要开辟通道，其中 bi<bi+1bi<bi+1，每两个整数之间用空格隔开（行尾没有空格）。

数据范围

2≤N,M≤10002≤N,M≤1000,
0≤K<M0≤K<M,
0≤L<N0≤L<N,
D≤2000D≤2000

输入样例：

4 5 1 2 3
4 2 4 3
2 3 3 3
2 5 2 4

输出样例：

2
2 4

解法1：

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;

/**
 * 坑：例子：结果是 9 4 1，那么一定要输出 1 4 9 ，没找到这种的快捷做法，于是就创了个temp数组来笨蛋做法
 */
public class Main {
    static int N = 1000 + 5;
    // row[1]维护1~2行的说话同学数目
    static Pair[] row = new Pair[N];
    static Pair[] col = new Pair[N];
    // 未排序之前就用pair储存

    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        String[] line = reader.readLine().split(" ");
        int m = Integer.parseInt(line[0]);
        int n = Integer.parseInt(line[1]);
        int k = Integer.parseInt(line[2]);
        int l = Integer.parseInt(line[3]);
        int d = Integer.parseInt(line[4]);
        for (int i = 0; i < N; i++) {
            row[i] = new Pair(i, 0);
            col[i] = new Pair(i, 0);
        }
        for (int i = 0; i < d; i++) {
            line = reader.readLine().split(" ");
            int x1 = Integer.parseInt(line[0]);
            int y1 = Integer.parseInt(line[1]);
            int x2 = Integer.parseInt(line[2]);
            int y2 = Integer.parseInt(line[3]);
            if (x1 == x2) {
                if (y1 < y2) {
                    col[y1].cnt++;
                } else {
                    col[y2].cnt++;
                }
            } else if (y1 == y2) {
                if (x1 < x2) {
                    row[x1].cnt++;
                } else {
                    row[x2].cnt++;
                }
            }
        }
        // 降序排列
        Arrays.sort(row, 1, m+1);
        Arrays.sort(col, 1, n+1);
        int[] temp=new int[N];
        //获取输出后对输出按字典序排序
        for (int i = 1; i <= k; i++) {
            temp[i]=row[i].x;
        }
        Arrays.sort(temp,1,k+1);
        for (int i = 1; i <= k - 1; i++) {
            System.out.print(temp[i] + " ");
        }
        if (k!=0) {
            System.out.print(temp[k]);
        }
        System.out.println();
        for (int i = 1; i <= l; i++) {
            temp[i]=col[i].x;
        }
        Arrays.sort(temp,1,l+1);
        for (int i = 1; i <= l - 1; i++) {
            System.out.print(temp[i] + " ");
        }
        if (l!=0) {
            System.out.print(temp[l]);
        }
    }
}

class Pair implements Comparable<Pair> {
    int x, cnt;

    Pair(int x, int cnt) {
        this.x = x;
        this.cnt = cnt;
    }

    // 降序
    @Override
    public int compareTo(Pair o) {
        // TODO Auto-generated method stub
        if (o.cnt!=this.cnt){
             return Integer.compare(o.cnt, this.cnt);
        }
       return Integer.compare(this.x,o.x);
    }

}