排座椅

题目描述

上课的时候总会有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的 𝐷 对同学上课时会交头接耳。

同学们在教室中坐成了 𝑀 行 𝑁 列，坐在第 𝑖 行第 𝑗 列的同学的位置是 (𝑖,𝑗)，为了方便同学们进出，在教室中设置了 𝐾 条横向的通道，𝐿 条纵向的通道。

于是，聪明的小雪想到了一个办法，或许可以减少上课时学生交头接耳的问题：她打算重新摆放桌椅，改变同学们桌椅间通道的位置，因为如果一条通道隔开了 2 个会交头接耳的同学，那么他们就不会交头接耳了。

请你帮忙给小雪编写一个程序，给出最好的通道划分方案。在该方案下，上课时交头接耳的学生的对数最少。

输入格式

第一行，有 5 个用空格隔开的整数，分别是 𝑀,𝑁,𝐾,𝐿,𝐷(2≤𝑁,𝑀≤1000,0≤𝐾<𝑀,0≤𝐿<𝑁,𝐷≤2000)。

接下来的 𝐷 行，每行有 4 个用空格隔开的整数。第 𝑖 行的 4 个整数 𝑋𝑖,𝑌𝑖,𝑃𝑖,𝑄𝑖，表示坐在位置 (𝑋𝑖,𝑌𝑖) 与 (𝑃𝑖,𝑄𝑖) 的两个同学会交头接耳（输入保证他们前后相邻或者左右相邻）。

输入数据保证最优方案的唯一性。

输出格式

共两行。
第一行包含 𝐾 个整数 𝑎1,𝑎2,…,𝑎𝐾​​，表示第 𝑎1​​ 行和 𝑎1+1 行之间、第 𝑎2​​ 行和 𝑎2+1 行之间、…、第 𝑎𝐾 行和第 𝑎𝐾+1 行之间要开辟通道，其中 𝑎𝑖<𝑎𝑖+1，每两个整数之间用空格隔开（行尾没有空格）。

第二行包含 𝐿 个整数 𝑏1,𝑏2,…,𝑏𝐿​​，表示第 𝑏1 列和 𝑏1+1 列之间、第 𝑏2 列和 𝑏2+1 列之间、…、第 𝑏𝐿​​ 列和第 𝑏𝐿+1 列之间要开辟通道，其中𝑏𝑖<𝑏𝑖+1​​，每两个整数之间用空格隔开（列尾没有空格）。

样例输入 #1

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

样例输出 #1

2
2 4

提示

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

AC代码

#include<bits/stdc++.h>
using namespace std;
const int N=5e3;
struct node{
	int v,l,r;
}a[N],b[N];
int m,n,k,L,d;
int x,y,p,q,t;
bool cmp1(node x,node y){
	return x.v>y.v;
}
bool cmp2(node x,node y){
	return x.l<y.l;
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0); 
	cin>>m>>n>>k>>L>>d;
	for(int i=1;i<m;i++){
		a[i].l=i,a[i].r=i+1;	
	}
	for(int i=1;i<n;i++){
		b[i].l=i,b[i].r=i+1;	
	}
	for(int i=1;i<=d;i++){
		cin>>x>>y>>p>>q;
		if(x==p){
			t=min(y,q);
			b[t].v++;
		}else{
			t=min(x,p);
			a[t].v++;
		}
	}
	sort(a+1,a+1+m-1,cmp1);
	sort(a+1,a+1+k,cmp2);
	sort(b+1,b+1+n-1,cmp1);
	sort(b+1,b+1+L,cmp2);
	for(int i=1;i<=k;i++) cout<<a[i].l<<" ";
	cout<<"\n";
	for(int i=1;i<=L;i++) cout<<b[i].l<<" ";
	return 0;
}