【C++题解】排座椅(seat)

题目链接：https://gmoj.net/junior/#main/show/1115

题目描述

上课的时候总有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的D对同学上课时会交头接耳。同学们在教室中坐成了M行N列，坐在第i行第j列的同学的位置是（i，j），为了方便同学们进出，在教室中设置了K条横向的通道，L条纵向的通道。于是，聪明的小雪想到了一个办法，或许可以减少上课时学生交头接耳的问题：她打算重新摆放桌椅，改变同学们桌椅间通道的位置，因为如果一条通道隔开了两个会交头接耳的同学，那么他们就不会交头接耳了。
请你帮忙给小雪编写一个程序，给出最好的通道划分方案。在该方案下，上课时交头接耳的学生对数最少。

输入

输出

输出文件seat.out共两行。
第一行包含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

数据范围限制

如题

提示

【输入输出样例解释】
上图中用符号*、※、+ 标出了3对会交头接耳的学生的位置，图中3条粗线的位置表示通道，图示的通道划分方案是唯一的最佳方案。

做法

一道模拟+贪心的题。

我们设 $lie[i].number$ 表示第 $i$ 列，$lie[i].people$ 表示第 $i$ 列会有多少同学交头接耳；设 $hang[i].number$ 表示第 $i$ 行， $hang[i].people$ 表示第 $i$ 行会有多少同学交头接耳。

当我们读入 $x_i,y_i,p_i,q_i$ 时，如果 $x_i=p_i$ ，说明第 $min(y_i,q_i)$ 列会有学生交头接耳，即让 $lie[min(y_i,q_i)].people$ 加 $1$ ；如果 $y_i=q_i$ ，说明第 $min(x_i,p_i)$ 行会有学生交头接耳，即让 $hang[min(x_i,p_i)].people$ 加 $1$ 。

题目要求在规定划分 $k$ 行和 $l$ 列的前提下，使得交头接耳的学生对数最少。即划分前 $k$ 大的 $hang$ 和前 $l$ 大的 $lie$ ，则从 $1$ 到 $m$ 让 $hang$ 从大到小排序，从 $1$ 到 $n$ 让 $lie$ 从大到小排序，就可以找出前 $k$ 大的 $hang$ 和前 $l$ 大的 $lie$ 了。

如图：

我们发现，输出的答案是严格从小到大的，则从 $1$ 到 $k$ 让 $hang$ 从小到大排序，从 $1$ 到 $l$ 让 $lie$ 从小到大排序。

排完序后，输出从 $1$ 到 $k$ 的 $hang$ 和从 $1$ 到 $l$ 的 $lie$ 即可。

#include<cstdio>
#include<algorithm>
#define min(a,b) (a<b?a:b)
using namespace std;
struct node{
	int number,poeple;
}hang[1005],lie[1005];
int m,n,k,l,d,x,y,p,q,cnt;
bool cmp(node a,node b){return a.poeple>b.poeple;}
bool cmp1(node a,node b){return a.number<b.number;}
int main()
{
	freopen("seat.in","r",stdin);
	freopen("seat.out","w",stdout);
	scanf("%d%d%d%d%d",&m,&n,&k,&l,&d);
	for(int i=1;i<=1000;i++)
		hang[i].number=lie[i].number=i;//lie[i].number表示第i列，hang[i].number表示第i行
	while(d--)
	{
		scanf("%d%d%d%d",&x,&y,&p,&q);
		if(x==p) lie[min(y,q)].poeple++;//说明第min(y_i,q_i)列会有学生交头接耳
		if(y==q) hang[min(x,p)].poeple++;//说明第min(x_i,p_i)行会有学生交头接耳
	}
	sort(hang+1,hang+m+1,cmp);//找出前k大的hang
	sort(hang+1,hang+k+1,cmp1);//输出要求，需要从小到大排序
	for(int i=1;i<=k;i++) printf("%d ",hang[i].number);
	printf("\n");
	sort(lie+1,lie+n+1,cmp);//找出前l大的lie
	sort(lie+1,lie+l+1,cmp1);//输出要求，需要从小到大排序
	for(int i=1;i<=l;i++) printf("%d ",lie[i].number);
	fclose(stdin);
	fclose(stdout);
	return 0;
}

写文章不易，望大家给 JG_DF_ 点个赞！