0
点赞
收藏
分享

微信扫一扫

小单刷题笔记之排座椅(贪心+排序)

夏木之下 2022-03-20 阅读 70

上课的时候总有一些同学和前后左右的人交头接耳,这是令小学班主任十分头疼的一件事情。不过,班主任小雪发现了一些有趣的现象,当同学们的座次确定下来之后,只有有限的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],每两个整数之间用空格隔开(行尾没有空格)。

输入输出样例

示例 1

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

2
2 4

思路:贪心+排序,一开始想的是读入两个小朋友的坐标就分一下,后来一想不是最优解。(但是骗了60的分,狗头保命) ,重新理一下思路,应该是按照分割的总次数来排序,每一次分割尽可量分开更多的熊孩子。难点就是处理这个按次数排序。用到一点桶排序的思想,然后按次数排序就行。

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn= 1005;
int M,N,K,L,D;
vector<int>a1,a2;
struct data
{
	int id,cnt;
	friend bool operator < (data a,data b)
	{
		return a.cnt>b.cnt;
	}
};
data a[maxn],b[maxn];
int main()
{
	cin>>M>>N>>K>>L>>D;
	for(int i=0;i<D;i++){
		int x1,y1,x2,y2;//x行数 y列数
		cin>>x1>>y1>>x2>>y2;
		if(x1==x2){//在同一行的两个小朋友
			b[min(y1,y2)].cnt++; //要记录这一分割可以拆散几对
                                
			b[min(y1,y2)].id=min(y1,y2);//如果在需要分的在相同列,不会改变要分的列数
		}
		if(y1==y2){
			a[min(x1,x2)].cnt++;
			a[min(x1,x2)].id=min(x1,x2);
		}
	}
	sort(a,a+maxn);//横着分割 按照每次分割可以分开的对数排序
	sort(b,b+maxn);//竖着分割 按照每次分割可以分开的对数排序
	
	for(int i=0;i<K;i++){
		if(a[i].id>0)a1.push_back(a[i].id);//a已经排好序了 把前K个元素插入a1
	}
	//sort(a1.begin(),a1.end());
	int lena=a1.size();
	for(int i=0;i<lena;i++){
		cout<<a1[i]<<" ";//插入的时候 就已经按照优先级排序了,按序输出就行
	}
	cout<<endl;
	
	
	for(int i=0;i<L;i++){
		if(b[i].id>0)a2.push_back(b[i].id);
	}
	sort(a2.begin(),a2.end());
	int lenb=a2.size();
	for(int i=0;i<lenb;i++){
		cout<<a2[i]<<" ";
	}
	

	
	return 0;
}
举报

相关推荐

0 条评论