洛谷 P1056 排座椅

题目链接

思路 : 结构体 + 排序 

(用结构体保存两个人说话的位置,接着去进行增加相同行或列的同学位置的次数,然后先筛选交头接耳的同学相同位置的次数由大到小的位置, 然后筛选前k个和前l个的位置就好) 

#include 
#include 
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

你可能感兴趣的:(~贪心)