题目描述
有一个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;
}