题目描述
上课的时候总会有一些同学和前后左右的人交头接耳,这是令小学班主任十分头疼的一件事情。不过,班主任小雪发现了一些有趣的现象,当同学们的座次确定下来之后,只有有限的D对同学上课时会交头接耳。
同学们在教室中坐成了M行N列,坐在第i行第j列的同学的位置是(i,j),为了方便同学们进出,在教室中设置了K条横向的通道,L条纵向的通道。
于是,聪明的小雪想到了一个办法,或许可以减少上课时学生交头接耳的问题:她打算重新摆放桌椅,改变同学们桌椅间通道的位置,因为如果一条通道隔开了2个会交头接耳的同学,那么他们就不会交头接耳了。
请你帮忙给小雪编写一个程序,给出最好的通道划分方案。在该方案下,上课时交头接耳的学生的对数最少。
输入输出格式
输入格式:
第一行,有5个用空格隔开的整数,分别是M,N,K,L,D(2<N,M<1000,0<K<M,0<L<N,D<2000)
接下来的D行,每行有4个用空格隔开的整数。第ii行的44个整数X_i,Y_i,P_i,Q_iXi,Yi,Pi,Qi,表示坐在位置(Xi,Yi)与(Pi,Qi)的两个同学会交头接耳(输入保证他们前后相邻或者左右相邻)。
输入数据保证最优方案的唯一性。
输出格式:
共两行。
第一行包含KK个整数a1,a2,…,aK,表示第a1行和a1+1行之间、第a−2行和a2+1行之间、…、第aK行和第aK+1行之间要开辟通道,其中ai<ai+1,每两个整数之间用空格隔开(行尾没有空格)。
第二行包含L个整数b1,b2,…,bL,表示第b1列和b1+1列之间、第b2列和b2+1列之间、…、第bL列和第bL+1列之间要开辟通道,其中bi<bi+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条粗线的位置表示通道,图示的通道划分方案是唯一的最佳方案。
思路
这道题就是个简单的贪心,但坑是真心多,尤其是输出格式中加粗的地方,很容易被阴。
代码
#include<bits/stdc++.h>
using namespace std;
struct node{
int x1,x2,y1,y2;
}a[2005];
bool cmp(int a,int b){
return a>b;
}
int k1[1005],l1[1005],l2[1005],k2[1005],ansk[1000],ansl[1000];
int n,m,k,l,d,z=1,y=1;
int main(){
//freopen("seat.in","r",stdin);
//freopen("seat.out","w",stdout);
cin>>n>>m>>k>>l>>d;
for(int i=1;i<=d;i++){
cin>>a[i].x1>>a[i].x2>>a[i].y1>>a[i].y2;
if(a[i].x2==a[i].y2) k1[min(a[i].x1,a[i].y1)]++;//这块可以对照图理解一下
else if(a[i].x1==a[i].y1) l1[min(a[i].x2,a[i].y2)]++;
}
memcpy(k2,k1,sizeof(k1));
memcpy(l2,l1,sizeof(l1));
sort(k1+1,k1+n,cmp);//排序
sort(l1+1,l1+m,cmp);
for(int i=1;i<=k;i++){//重新匹配
for(int j=1;j<n;j++){
if(k2[j]==k1[i]){
k2[j]=0;//去重
ansk[i]=j;
break;
}
}
}
for(int i=1;i<=l;i++){//重新匹配
for(int j=1;j<m;j++){
if(l2[j]==l1[i]){
l2[j]=0;//去重
ansl[i]=j;
break;
}
}
}
sort(ansk+1,ansk+k+1);//特别坑,输出要按顺序
sort(ansl+1,ansl+l+1);
for(int i=1;i<=k;i++){
cout<<ansk[i]<<" ";
}
cout<<endl;
for(int i=1;i<=l;i++){
cout<<ansl[i]<<" ";
}
return 0;
}/*
4 5 1 2 3
4 2 4 3
2 3 3 3
2 5 2 4
*/