排座椅 \operatorname{排座椅} 排座椅
题目链接: luogu P1056 \operatorname{luogu\ P1056} luogu P1056
题目
上课的时候总会有一些同学和前后左右的人交头接耳,这是令小学班主任十分头疼的一件事情。不过,班主任小雪发现了一些有趣的现象,当同学们的座次确定下来之后,只有有限的 D D D 对同学上课时会交头接耳。
同学们在教室中坐成了 M M M 行 N N N 列,坐在第 i i i 行第 j j j 列的同学的位置是 ( i , j ) (i,j) (i,j) ,为了方便同学们进出,在教室中设置了 K K K 条横向的通道, L L L 条纵向的通道。
于是,聪明的小雪想到了一个办法,或许可以减少上课时学生交头接耳的问题:她打算重新摆放桌椅,改变同学们桌椅间通道的位置,因为如果一条通道隔开了 2 2 2 个会交头接耳的同学,那么他们就不会交头接耳了。
请你帮忙给小雪编写一个程序,给出最好的通道划分方案。在该方案下,上课时交头接耳的学生的对数最少。
输入
第一行,有 5 5 5 个用空格隔开的整数,分别是 M , N , K , L , D ( 2 ≤ N , M ≤ 1000 , 0 ≤ K < M , 0 ≤ L < N , D ≤ 2000 ) M,N,K,L,D(2 \le N,M \le 1000,0 \le K<M,0 \le L<N,D \le 2000) M,N,K,L,D(2≤N,M≤1000,0≤K<M,0≤L<N,D≤2000) 。
接下来的 D D D 行,每行有 4 4 4 个用空格隔开的整数。第 i i i 行的 4 4 4 个整数 X i , Y i , P i , Q i X_i,Y_i,P_i,Q_i Xi,Yi,Pi,Qi ,表示坐在位置 ( X i , Y i ) (X_i,Y_i) (Xi,Yi) 与 ( P i , Q i ) (P_i,Q_i) (Pi,Qi) 的两个同学会交头接耳(输入保证他们前后相邻或者左右相邻)。
输入数据保证最优方案的唯一性。
输出
共两行。
第一行包含
K
K
K 个整数
a
1
,
a
2
,
…
,
a
K
a_1,a_2,\ldots,a_K
a1,a2,…,aK ,表示第
a
1
a_1
a1 行和
a
1
+
1
a_1+1
a1+1 行之间、第
a
2
a_2
a2 行和
a
2
+
1
a_2+1
a2+1 行之间、
…
…
… 、第
a
K
a_K
aK 行和第
a
K
+
1
a_K+1
aK+1 行之间要开辟通道,其中
a
i
<
a
i
+
1
a_i< a_{i+1}
ai<ai+1 ,每两个整数之间用空格隔开(行尾没有空格)。
第二行包含 L L L 个整数 b 1 , b 2 , … , b L b_1,b_2,\ldots,b_L b1,b2,…,bL ,表示第 b 1 b_1 b1 列和 b 1 + 1 b_1+1 b1+1 列之间、第 b 2 b_2 b2 列和 b 2 + 1 b_2+1 b2+1 列之间、 … … … 、第 b L b_L bL 列和第 b L + 1 b_L+1 bL+1 列之间要开辟通道,其中 b i < b i + 1 b_i< b_{i+1} bi<bi+1 ,每两个整数之间用空格隔开(列尾没有空格)。
样例输入
4 5 1 2 3
4 2 4 3
2 3 3 3
2 5 2 4
样例输出
2
2 4
样例解释
上图中用符号*、※、+标出了
3
3
3 对会交头接耳的学生的位置,图中
3
3
3 条粗线的位置表示通道,图示的通道划分方案是唯一的最佳方案。
思路
这道题是一道贪心。
我们可以发现,其实左右的和上下的不会干扰,分别只能用竖的和横的来分割。
那我们就分别处理。
就排序找到能分开更多 cp 的道路,选前
k
/
l
k/l
k/l 个就是了。
有一个很重要的点就是,输出的道路也要排序,从小到大输出。
(不然只有
10
10
10 分)
代码
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
struct node{
int x, num;
}shu[2001], heng[2001];
int m, n, k, l, d, x1, x2, y1, y2, ans1[2001], ans2[2001];
int read() {//快读
int an = 0, zhengfu = 1;
char c = getchar();
while (c < '0' || c > '9') {
if (c == '-') zhengfu = -zhengfu;
c = getchar();
}
while (c >= '0' && c <= '9') {
an = an * 10 + c - '0';
c = getchar();
}
return an;
}
bool cmp(node x, node y) {//按可以拆开的个数从大到小排序
return x.x > y.x;
}
int main() {
n = read();//读入
m = read();
k = read();
l = read();
d = read();
for (int i = 1; i <= max(n, m); i++) {
if (i < n) heng[i].num = i;//初始化
if (i < m) shu[i].num = i;
}
for (int i = 1; i <= d; i++) {
x1 = read();//读入
y1 = read();
x2 = read();
y2 = read();
if (x1 == x2) {
shu[min(y1, y2)].x++;//是左右cp
}
else {
heng[min(x1, x2)].x++;//是前后cp
}
}
sort(shu + 1, shu + m + 1, cmp);//按可以拆开的个数从大到小排序
sort(heng + 1, heng + n + 1, cmp);
for (int i = 1; i <= k; i++)
ans1[++ans1[0]] = heng[i].num;//找到最佳的分割位置
for (int i = 1; i <= l; i++)
ans2[++ans2[0]] = shu[i].num;
sort(ans1 + 1, ans1 + ans1[0] + 1);//要排序
sort(ans2 + 1, ans2 + ans2[0] + 1);
for (int i = 1; i <= ans1[0]; i++)//输出
printf("%d ", ans1[i]);
printf("\n");
for (int i = 1; i <= ans2[0]; i++)
printf("%d ", ans2[i]);
return 0;
}