https://ac.nowcoder.com/acm/contest/20960/1026
用两个数组分别表示每一行、列可以隔开多少对爱说话的小朋友即可,由于按编号升序输出,因此先找出行列所在的位置再按编号进行排序即可。
#include<bits/stdc++.h>
#define endl '\n'
#define pii pair<int,int>
using namespace std;
const int maxn = 1003;
struct order
{
int num,id;
}hang[maxn],lie[maxn];
bool cmp1(order x,order y)
{
return x.num>y.num;
}
bool cmp2(order x,order y)
{
return x.id<y.id;
}
void solve()
{
int n,m,k,l,d;
cin>>n>>m>>k>>l>>d;
int x,y,p,q;
for(int i=1;i<=n;i++)
hang[i].id=i;
for(int i=1;i<=m;i++)
lie[i].id=i;
while(d--)
{
cin>>x>>y>>p>>q;
if(x==p) // 同一行
{
int temp=min(y,q);
lie[temp].num++;
}
else
{
int temp=min(x,p);
hang[temp].num++;
}
}
sort(hang+1,hang+1+n,cmp1);
sort(lie+1,lie+1+m,cmp1);
sort(hang+1,hang+1+k,cmp2);
sort(lie+1,lie+1+l,cmp2);
for(int i=1;i<=k;i++)
cout<<hang[i].id<<" ";
cout<<endl;
for(int i=1;i<=l;i++)
cout<<lie[i].id<<" ";
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int T; T=1;
while(T--) solve();
return 0;
}