全国 2008 NOIP 普及组试题 排座椅

一、题目

题目描述 Description

上课的时候总会有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的D对同学上课时会交头接耳。同学们在教室中坐成了M行N列，坐在第i行第j列的同学的位置是（i,j），为了方便同学们进出，在教室中设置了K条横向的通道，L条纵向的通道。于是，聪明的小雪想到了一个办法，或许可以减少上课时学生交头接耳的问题：她打算重新摆放桌椅，改变同学们桌椅间通道的位置，因为如果一条通道隔开了两个会交头接耳的同学，那么他们就不会交头接耳了。

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

输入描述 Input Description

输入格式：
输入文件seat.in的第一行，有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)的两个同学会交头接耳（输入保证他们前后相邻或者左右相邻）。

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

输出描述 Output Description

输出格式：
输出文件seat.out共两行。

第一行包含K个整数，a1，a2……aK，表示第a1行和a1+1行之间、第a2行和a2+1行之间、…、第aK行和第aK+1行之间要开辟通道，其中ai< ai+1，每两个整数之间用空格隔开（行尾没有空格）。

第二行包含L个整数，b1，b2……bL，表示第b1列和b1+1列之间、第b2列和b2+1列之间、…、第bL列和第bL+1列之间要开辟通道，其中bi< bi+1，每两个整数之间用空格隔开（列尾没有空格）。

样例输入 Sample Input

输入样例#1： 4 5 1 2 3 4 2 4 3 2 3 3 3 2 5 2 4

样例输出 Sample Output

输出样例#1： 2 2 4

数据范围及提示 Data Size & Hint

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

二、思路

        所有的通道一定划分在交头接耳说话的一对或者多对学生之间。而现在只能划分固定的通道次数，行方向上可以划分出K条通道，列方向上可以划分出L条通道。所以需要保证每一条通道划分的学生数量都尽可能多。

        具体写法是：

                先统计出每两行和每两列之间交头接耳说话的学生数量，将数量和行号（或列好）都分别存入两个数组

                然后对数组按学生数量分别从小到大排序

                在行方向是挑出学生数量最多的k个，列方向上挑出学生数量最多的l个，然后再按照行号或者列号格子从小到大排序

                最后输出即可

三、完整代码

#include <bits/stdc++.h>     

/*个人习惯用万能头，其实需要#include <iostream>和#include <algorithm> */

using namespace std;

struct node {

        int id,s;

};

int m,n,k,l,d;

node a[1005],b[1005];

bool cmp(node a,node b){

        return a.s>b.s;

}

bool cmp2(node a,node b) {

        return a.id<b.id;

}

int main(){

        cin>>m>>n>>k>>l>>d;       

        for(int i=1;i<=d;i++){

                int x1,y1,x2,y2;

                cin>>x1>>y1>>x2>>y2;

                if(x1==x2) {        //y1,y2

                        b[min(y1,y2)].s++;        //桶

                        b[min(y1,y2)].id=min(y1,y2);

                 }else if(y1==y2) {

                        a[min(x1,x2)].s ++;//桶

                        a[min(x1,x2)].id = min(x1,x2);

                }

        }
    //a[i]:第i行和第i+1行说话的人数
    //b[i]:第i列和第i+1列说话的人数

        sort(a+1,a+m+1,cmp); 

        sort(a+1,a+k+1,cmp2);

        sort(b+1,b+n+1,cmp);

        sort(b+1,b+l+1,cmp2)；

        for(int i=1; i<=k; i++）cout<<a[i].id<<' ';

        cout<<endl;

        for(int i=1;i<=l;i++）cout<<b[i].id<<' ';

        return 0;
}