咨询了一波涵大佬的之后,运用了一波set这个数据结构
set排进去的数据,有序,并且可以去重,正好可以用于本题!!
本题主要思路:找到哪些列(行)可以分开的人最多,选择前多少个(题目要求)这样的行(列)的编号,输出就可以了,主要是理解题意,贪心思想
题目描述
上课的时候总会有一些同学和前后左右的人交头接耳,这是令小学班主任十分头疼的一件事情。不过,班主任小雪发现了一些有趣的现象,当同学们的座次确定下来之后,只有有限的D对同学上课时会交头接耳。
同学们在教室中坐成了MM行NN列,坐在第i行第j列的同学的位置是(i,j)(i,j),为了方便同学们进出,在教室中设置了KK条横向的通道,LL条纵向的通道。
于是,聪明的小雪想到了一个办法,或许可以减少上课时学生交头接耳的问题:她打算重新摆放桌椅,改变同学们桌椅间通道的位置,因为如果一条通道隔开了22个会交头接耳的同学,那么他们就不会交头接耳了。
请你帮忙给小雪编写一个程序,给出最好的通道划分方案。在该方案下,上课时交头接耳的学生的对数最少。
输入格式
第一行,有55个用空格隔开的整数,分别是M,N,K,L,D(2 \le N,M \le 1000,0 \le K<M,0 \le L<N,D \le 2000)M,N,K,L,D(2≤N,M≤1000,0≤K<M,0≤L<N,D≤2000)
接下来的DD行,每行有44个用空格隔开的整数。第ii行的44个整数X_i,Y_i,P_i,Q_iXi,Yi,Pi,Qi,表示坐在位置(X_i,Y_i)(Xi,Yi)与(P_i,Q_i)(Pi,Qi)的两个同学会交头接耳(输入保证他们前后相邻或者左右相邻)。
输入数据保证最优方案的唯一性。
输出格式
共两行。
第一行包含KK个整数a_1,a_2,…,a_Ka1,a2,…,aK,表示第a_1a1行和a_1+1a1+1行之间、第a_2a2行和a_2+1a2+1行之间、…、第a_KaK行和第a_K+1aK+1行之间要开辟通道,其中a_i< a_i+1ai<ai+1,每两个整数之间用空格隔开(行尾没有空格)。
第二行包含LL个整数b_1,b_2,…,b_Lb1,b2,…,bL,表示第b_1b1列和b_1+1b1+1列之间、第b_2b2列和b_2+1b2+1列之间、…、第b_LbL列和第b_L+1bL+1列之间要开辟通道,其中b_i< b_i+1bi<bi+1,每两个整数之间用空格隔开(列尾没有空格)。
输入
4 5 1 2 3 4 2 4 3 2 3 3 3 2 5 2 4
输出
2 2 4
说明/提示
上图中用符号*、※、+标出了33对会交头接耳的学生的位置,图中33条粗线的位置表示通道,图示的通道划分方案是唯一的最佳方案
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<set>
#include<stack>
using namespace std;
int sx,sy,fx,fy;
int n,m,k,l,d;
typedef struct
{
int id;
int val;//可以分开多少对人
}Node;
bool cmp(Node a,Node b)
{
return a.val>b.val;
}
Node nodeh[2009];//行
Node nodeli[2009];//列
set<int>s1;
set<int>s2;
int main()
{
scanf("%d %d %d %d %d",&n,&m,&k,&l,&d);
for(int i=1;i<=n;i++)
nodeh[i].id=i;
for(int i=1;i<=m;i++)
nodeli[i].id=i;
for(int i=1;i<=d;i++)
{
scanf("%d %d %d %d",&sx,&sy,&fx,&fy);
if(sx==fx)
nodeli[(sy+fy)>>1].val++;
else if(fy==sy)
nodeh[(sx+fx)>>1].val++;
}
sort(nodeh+1,nodeh+1+n,cmp);
sort(nodeli+1,nodeli+1+m,cmp);
for(int i=1;i<=k;i++)
s1.insert(nodeh[i].id);
for(int i=1;i<=l;i++)
s2.insert(nodeli[i].id);
set<int>::iterator it;
for(it=s1.begin();it!=s1.end();it++)
printf("%d ",*it);
printf("\n");
for(it=s2.begin();it!=s2.end();it++)
printf("%d ",*it);
return 0;
}