NC16618 [NOIP2008]排座椅

题目

题目描述

上课的时候总有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的 \(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\) 个整数 \(X_i,Y_i,P_i,Q_i\)，表示坐在位置 \((X_i,Y_i)\) 与 \((P_i,Q_i)\) 的两个同学会交头接耳（输入保证他们前后相邻或者左右相邻）。

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

输出描述

共两行
第一行包含 \(K\) 个整数，\(a_1a_2……a_K\)，表示第 \(a_1\) 行和 \(a_1+1\) 行之间、第 \(a_2\) 行和第 \(a_2+1\) 行之间、…、第 \(a_K\) 行和第 \(a_K+1\) 行之间要开辟通道，其中 \(a_i < a_{i+1}\)，每两个整数之间用空格隔开（行尾没有空格）。
第二行包含 \(L\) 个整数，\(b_1b_2……b_k\)，表示第 \(b_1\) 列和 \(b_1+1\) 列之间、第 \(b_2\) 列和第 \(b_2+1\) 列之间、…、第 \(b_L\) 列和第 \(b_L+1\) 列之间要开辟通道，其中 \(b_i< b_{i+1}\)，每两个整数之间用空格隔开（行尾没有空格）。

示例1

输入

4 5 1 2 3
4 2 4 3
2 3 3 3
2 5 2 4

输出

2
2 4

题解

知识点：贪心。

显然，行和列的分布是独立的。记录每行（列）的对数，从大到小排序，取前 \(K\) (\(L\)) 个再按照行列数排序即可。

代码

#include <bits/stdc++.h>

using namespace std;

pair<int, int> r[1007], c[1007];

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int m, n, k, l, d;
    cin >> m >> n >> k >> l >> d;
    for (int i = 1;i < m;i++) r[i].second = i, r[i].first = 0;
    for (int i = 1;i < n;i++) c[i].second = i, c[i].first = 0;
    while (d--) {
        int x1, y1, x2, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        if (x1 == x2) c[min(y1, y2)].first++;
        else if (y1 == y2) r[min(x1, x2)].first++;
    }
    sort(r + 1, r + m, [&](pair<int, int> a, pair<int, int> b) {return a.first > b.first;});
    sort(c + 1, c + n, [&](pair<int, int> a, pair<int, int> b) {return a.first > b.first;});
    sort(r + 1, r + k + 1, [&](pair<int, int> a, pair<int, int> b) {return a.second < b.second;});
    sort(c + 1, c + l + 1, [&](pair<int, int> a, pair<int, int> b) {return a.second < b.second;});
    for (int i = 1;i <= k;i++) cout << r[i].second << " ";
    cout << '\n';
    for (int i = 1;i <= l;i++) cout << c[i].second << " ";
    cout << '\n';
    return 0;
}