CCF NOI 1055.走廊

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

输入
第一行，有5各用空格隔开的整数，分别是M，N，K，L，D（2<=N，M<=1000，0<=K

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int M, N, K, L, D, rx[2001], ry[2001], x1, y1, x2, y2;
struct Data {
    int id, val;//p是行列号，val是权重
}x[1001], y[1001];
bool cmp(Data a, Data b) {//按照权重排序
    return a.val>b.val;
}
int main() {
    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;
}