题目链接
思路 : 结构体 + 排序
(用结构体保存两个人说话的位置,接着去进行增加相同行或列的同学位置的次数,然后先筛选交头接耳的同学相同位置的次数由大到小的位置, 然后筛选前k个和前l个的位置就好)
#include <iostream>
#include <algorithm>
using namespace std;
struct Node{
int index, v = 0; //index是下标, v保存最多的两个人中坐标较小说话的那一行或者那一列的次数
}ha[1005], li[1005];
bool cmp1(Node a, Node b) //降序
{
return a.v > b.v;
}
bool cmp2(Node a, Node b) //升序
{
return a.index < b.index;
}
int main()
{
int m, n, l, k, d, x, y, p, q;
while(cin >> m >> n >> k >> l >> d)
{
for(int i = 0; i <= m; i++) //初始化下标
ha[i].index = i;
for(int i = 0; i <= n; i++)
li[i].index = i;
for(int i = 0; i < d; i++)
{
cin >> x >> y >> p >> q;
if(x == p) //当行相同时选择两个坐标中列较小的v++;
y > q ? li[q].v++ : li[y].v++;
else //同理
x > p ? ha[p].v++ : ha[x].v++;
}
sort(ha + 1, ha + m + 1, cmp1); //升序,因为要输出最优解
sort(li + 1, li + n + 1, cmp1);
sort(ha + 1, ha + k + 1, cmp2); //位置降序因为结果要从小到大进行输出
sort(li + 1, li + l + 1, cmp2);
for(int i = 1; i <= k; i++)
{
cout << ha[i].index;
if(i != k)
cout << " ";
}
cout << endl;
for(int i = 1; i <= l; i++)
{
cout << li[i].index;
if(i != l)
cout << " ";
}
cout << endl;
}
return 0;
}
参考博客https://blog.csdn.net/Axone/article/details/80866544