【一只蒟蒻的刷题历程】 【洛谷】 P1056-排座椅 （贪心思想）

题目描述

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

同学们在教室中坐成了 M 行 N 列，坐在第 ii 行第 jj 列的同学的位置是 (i,j)，为了方便同学们进出，在教室中设置了 K 条横向的通道，L 条纵向的通道。

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

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

输入格式

输出格式

输入输出样例

输入 #1

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

输出 #1

2
2 4

说明/提示

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

2008 年普及组第二题

---

思路：

贪心思想：每使用一条通道要使被隔开的学生尽可能多。

因此，首先记录每行和每列可以隔开多少个学生，然后进行排序，能隔开越多学生的行和列要优先设立通道。最后把这些 设立的通道的行号和列号进行排序，按要求输出。

ps：map的value不能直接排序，需要先放入pair里，再放入vector里

---

代码：

#include <iostream>
#include <algorithm>
#include <string>
#include <queue>
#include <cstring>
#include <cmath>
#include <vector>
#include <set>
#include <map>
using namespace std;  
typedef pair<int,int> pai;

bool cmp(pai a,pai b){  //排序，pair的second就是可以隔开的学生数
	return a.second>b.second;
}

int m,n,k,l,d,x,y,p,q;
map<int,int> row,col; //记录可以隔开的学生数
vector<pai> v1,v2;
vector<int> ans1,ans2;
int main() 
{
	cin>>m>>n>>k>>l>>d;
	for(int i=0;i<d;i++)
	{
	    cin>>x>>y>>p>>q; 
		if(x==p) col[min(y,q)]++; 
		else if(y==q)  row[min(x,p)]++;
	}
	for(int i=0;i<=1000;i++)  //(行/列，隔开数) 装入pair类型的vector里
	{
		v1.push_back({i,row[i]});
		v2.push_back({i,col[i]});
	}
	
	sort(v1.begin(),v1.end(),cmp); //隔开数排序
	sort(v2.begin(),v2.end(),cmp);
	
	for(int i=0;i<k;i++) ans1.push_back(v1[i].first); //前k个行号
	for(int i=0;i<l;i++) ans2.push_back(v2[i].first); //前l个列号

	sort(ans1.begin(),ans1.end()); //行列号排序
	sort(ans2.begin(),ans2.end());
	
	for(int i=0;i<ans1.size();i++){ //按要求输出
		cout<<ans1[i];
		if(i<ans1.size()-1) cout<<" ";
	}
	cout<<endl;
	
	for(int i=0;i<ans2.size();i++){
		cout<<ans2[i];
		if(i<ans2.size()-1) cout<<" ";
	}
	cout<<endl;
	return 0;
}