1055. 走廊

题目描述：

有一个M行N列的教室座位中，有D对同学总爱凑在一起讲话。现老师要用走廊隔开他们。但只能在行之间加入K条走廊，在列中加入L条走廊，问加在哪里能使效果最佳。（一对爱讲话的同学只有左右相邻或上下相邻）。

输入

第一行，有5各用空格隔开的整数，分别是M，N，K，L，D（2<=N，M<=1000，0<=K < M，0<=L< N，D<=2000）。接下来D行，每行有4个用空格隔开的整数，第i行的4个整数Xi，Yi，Pi，Qi，表示坐在位置(Xi,Yi)与(Pi,Qi)的两个同学会交头接耳（输入保证他们前后相邻或者左右相邻）。输入数据保证最优方案的唯一性。

输出：

共两行。第一行包含K个整数，a1a2…aK，表示第a1行和a1+1行之间、第a2行和第a2+1行之间、…、第aK行和第aK+1行之间要开辟通道，其中ai< ai+1，每两个整数之间用空格隔开（行尾没有空格）。第二行包含L个整数，b1b2…bk，表示第b1列和b1+1列之间、第b2列和第b2+1列之间、…、第bL列和第bL+1列之间要开辟通道，其中bi< bi+1，每两个整数之间用空格隔开（行尾没有空格）。若有多组答案，输出字典序最小的一组。

样例输入

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

样例输出

2
2 4

代码如下：

#include<bits/stdc++.h>
using namespace std;
int M, N, K, L, D, rx[2001], ry[2001]; 
struct Data {
    int id, val;//id是行列号，val是权重
}x[1001], y[1001];
bool cmp(Data a, Data b) {//按照权重排序
    if(a.val==b.val)
    return a.id<b.id;
    return a.val>b.val;
}
int main() {
    int x1, y1, x2, y2;
    cin >> M >> N >> K >> L >> D;
    for (int i = 1; i <= 1001; i++) {
        y[i].id = i;//对行列号赋值，因为后面排序会打乱顺序
    }
    for (int i = 1; i <= 1001; i++) {
        x[i].id = i;
    }
    for (int i = 1; i <= D; i++) {
        cin >> x1 >> y1 >> x2 >> y2;
        if (y1 == y2) {
            x[min(x1, x2)].val++;//如果在同一列那min(x1,x2)这一行价值加一
        }
        else if (x1 == x2)
        {
            y[min(y1, y2)].val++;//如果在同一行那min(y1,y2)这一列价值加一
        }
    }
    sort(x + 1, x + 1001, cmp);//按照权重及行列号排序
    sort(y + 1, y + 1001, cmp);
    for (int i = 1; i <= K; i++) {
        rx[i] = x[i].id;//在价值最大的几列里面取出要求量的列并储存列号
    }
    for (int i = 1; i <= L; i++) {
        ry[i] = y[i].id;//在价值最大的几行里面取出要求量的行并储存行号
    }
    sort(rx + 1, rx + 1 + K);//对列号进行排序
    sort(ry + 1, ry + 1 + L);//对行号进行排序
    for (int i = 1; i <= K; i++) {
        cout << rx[i] << " ";
    }
    cout << endl;
    for (int i = 1; i <= L; i++) {
        cout << ry[i] << " ";
    }
    return 0;
}