排座位(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 << "\n"; 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; }