普及组2008NOIP 排座椅(贪心+排序）

转载

蔡军帅 2021-09-06 11:27:13

文章标签 i++ #include 数据 ios 升序 文章分类 数据结构与算法人工智能

排座椅

时间限制: 1 Sec 内存限制: 50 MB

题目描述

上课的时候总有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的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个整数，a1 a2...aK，表示第a1行和a1+1行之间、第a2行和第a2+1行之间、...、第aK行和第aK+1行之间要开辟通道，其中ai<ai+1，每两个整数之间用空格隔开（行尾没有空格）。

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

下面是对样例数据的解释：

普及组2008NOIP 排座椅(贪心+排序）_#include

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

样例输入

样例输出

2
2 4

提示

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
 
struct edge
{
    int num;
    int sum;
};
 
bool cmp1(edge a,edge b)
{
    return a.sum > b.sum;
}
 
bool cmp2(edge a,edge b)
{
    return a.num < b.num;
}
 
edge row[2001];
edge col[2001];
 
int main()
{
    int M,N,K,L,D;
    while(cin >> M )
    {
        cin >> N >> K >> L >> D;
        int i;
        for(i = 0;i < M;i++)
        {
            row[i].num = i + 1;
            row[i].sum = 0;
        }
        for(i = 0;i < N;i++)
        {
            col[i].num = i + 1;
            col[i].sum = 0;
        }
        for(i = 0;i < D;i++)
        {
            int a,b,x,y;
            cin >> a >> b >> x >> y;
            if(a == x)
            {
                col[min(b,y) - 1].sum++;;
            }
            if(b == y)
            {
                row[min(a,x) - 1].sum++;
            }
        }
        //陷阱一:注意号码要升序输出，注意要再排一次序
        sort(row,row+M,cmp1);
        sort(col,col+N,cmp1);
        sort(row,row+K,cmp2);
        sort(col,col+L,cmp2);
        //注意:最后一项后面不必输出空格
        for(i = 0;i < K;i++)
        {
            if(i != K - 1)
                cout << row[i].num << ' ';
            else
                cout << row[i].num;
        }
        cout << endl;
 
        for(i = 0;i < L;i++)
        {
            if(i != L - 1)
                cout << col[i].num << ' ';
            else
                cout << col[i].num;
        }
        cout << endl;
    }
    return 0;
}

排座椅