【问题描述】
上课的时候总有一些同学和前后左右的人交头接耳,这是令小学班主任十分头疼的一件事情。不过,班主任小雪发现了一些有趣的现象,当同学们的座次确定下来之后,只有有限的D对同学上课时会交头接耳。同学们在教室中坐成了M行N列,坐在第i行第j列的同学的位置是(i,j),为了方便同学们进出,在教室中设置了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个整数,a[1] a[2] … a[K],表示第a[1]行和a[1+1]行之间、第a[2]行和第a[2+1]行之间、…、第a[K]行和第a[K+1]行之间要开辟通道,其中a[i]< a[i+1],每两个整数之间用空格隔开(行尾没有空格)。
第二行包含L个整数,b[1] b[2] … b[k],表示第b[1]列和b[1+1]列之间、第b[2]列和第b[2+1]列之间、…、第b[L]列和第b[L+1]列之间要开辟通道,其中b[i]< b[i+1],每两个整数之间用空格隔开(行尾没有空格)。
【输入样例】
4 5 1 2 3
4 2 4 3
2 3 3 3
2 5 2 4
【输出样例】
2
2 4
【样例解释】
上图中用符号*、※、+ 标出了3对会交头接耳的学生的位置,图中3条粗线的位置表示通道,图示的通道划分方案是唯一的最佳方案。
计算每一排和每一列的讲话对数,贪心优先隔离讲话对数多的排/列。记得最后要把答案由编号从小到大排序,否则酿成大祸。(血与泪的教训)
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 2502;
int m, n, k, l, d;
//计算每一排和每一列的讲话对数,贪心优先隔离讲话对数多的排/列
struct data
{
int w, id;
};
data totx[maxn], toty[maxn]; //totx横向讲话 ,toty纵向讲话
bool cmp(data a, data b)
{
if (a.w == b.w) return a.id<b.id;
return a.w>b.w;
}
vector<int>x, y;
int main()
{
//freopen("1.txt", "r", stdin);
//freopen("seat.out", "w", stdout);
scanf("%d%d%d%d%d", &m, &n, &k, &l, &d);
memset(totx, 0, sizeof(totx));
memset(toty, 0, sizeof(toty));
for (int i = 1; i <= d; i++)
{
int x1, y1, p1, q1;
scanf("%d %d %d %d", &x1, &y1, &p1, &q1);
if (x1 == p1) //纵向讲话
{
if (y1>q1) toty[q1].w++, toty[q1].id = q1;
else toty[y1].w++, toty[y1].id = y1;
}
if (q1 == y1) //横向讲话
{
if (x1>p1) totx[p1].w++, totx[p1].id = p1;
else totx[x1].w++, totx[x1].id = x1;
}
}
sort(totx + 1, totx + m + 1, cmp);
for (int i = 1; i <= k; i++) //横向讲话隔离
{
x.push_back(totx[i].id);
}
sort(toty + 1, toty + n + 1, cmp);
for (int i = 1; i <= l; i++) //纵向讲话隔离
{
y.push_back(toty[i].id);
}
sort(x.begin(), x.end());
sort(y.begin(), y.end());
for (int i = 0; i < k; i++)
{
printf("%d ", x[i]);
}
printf("\n");
for (int i = 0; i < l; i++)
{
printf("%d ", y[i]);
}
return 0;
}