排座位（newcoder）

 题目链接：https://ac.nowcoder.com/acm/problem/16618

 题目描述

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

示例

输入：
4 5 1 2 3
4 2 4 3
2 3 3 3
2 5 2 4

输出：
2 
2 4

本人代码：

（未acc，只满足对D组数据的分配）

#include<iostream>
#include<queue>
#include<algorithm>
#include<unordered_map>
using namespace std;
struct point
{
    int x;
    int y;
}a[2];
int main()
{
    ios::sync_with_stdio(false);
    unordered_map<int, int>v1;
    unordered_map<int, int>h1;
    priority_queue<pair<int, int>>v;
    priority_queue<pair<int, int>>h;
    int m, n, k, l, d;
    vector<int>ans;
    cin >> m >> n >> k >> l >> d;
    for (int z = 0;z < d;z++)
    {
        cin >> a[0].x >> a[0].y >> a[1].x >> a[1].y;
        if (a[0].x == a[1].x)
            v1[min(a[0].y, a[1].y)]++;
        else
            h1[min(a[0].x, a[1].x)]++;
    }
    for (auto x : v1)
        v.push(make_pair(x.second, x.first));
    for (auto x : h1)
        h.push(make_pair(x.second, x.first));
    int f = 0;
    while (true)
    {
        ans.push_back(v.top().second);
        v.pop();
        f++;
        if (f == l)
            break;
    }
    sort(ans.begin(), ans.end());
    for (auto x : ans)
        cout << x << " ";
    cout << "
";
    ans.clear();
    f = 0;
    while (true)
    {
        ans.push_back(h.top().second);
        h.pop();
        f++;
        if (f == k)
            break;
    }
    sort(ans.begin(), ans.end());
    for (auto x : ans)
        cout << x << " ";
}

附上神仙代码：

#include<iostream>
#include<algorithm>
using namespace std;
struct ty {
    int num;
    int cnt;
} horizontal[1001], vertical[1001];
bool cmp1(ty x, ty y) 
{
    return x.cnt > y.cnt;
}
bool cmp2(ty x, ty y) 
{
    return x.num < y.num;
}
int main()
{
    ios::sync_with_stdio(false);
    int m, n, k, l, d;
    cin >> m >> n >> k >> l >> d;
    for (int i = 1;i <= m;i++)
        horizontal[i].num = i;
    for (int i = 1;i <= n;i++)
        vertical[i].num = i;
    for (int i = 1;i <= d;i++) 
    {
        int a1, b1, a2, b2;
        cin >> a1 >> b1 >> a2 >> b2;
        if (a1 != a2) horizontal[min(a1, a2)].cnt++;
        else vertical[min(b1, b2)].cnt++;
    }
    sort(horizontal + 1, horizontal + m, cmp1);
    sort(vertical + 1, vertical + n, cmp1);

    sort(horizontal + 1, horizontal + 1 + k, cmp2);
    sort(vertical + 1, vertical + 1 + l, cmp2);

    for (int i = 1;i <= k;i++) 
    {
        printf("%d ", horizontal[i].num);
    }
    cout << endl;
    for (int i = 1;i <= l;i++) 
    {
        printf("%d ", vertical[i].num);
    }
    return 0;
}