Codeforces 1148D. Dirty Deeds Done Dirt Cheap
传送门:https://codeforces.com/problemset/problem/1148/D
提醒自己:有时pair可以发挥比较好的作用,用pair自带序号可能会比find快。
主要思路:
先把数据分为两类,一类为a大于b型,另一类为b大于a型,显然,如果要符合条件,所选中的数据必为同一种类型。然后比较两种数据的个数多少,决定使用哪一种数据,易知,如果ai>bi,aj>ai则aj>bi所以,只用针对不同情况的a或b排序即可。因为只要序号,所以不用管当a排序后b是否还要排序。
代码如下:
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n,x=0,y=0,z;
cin >>n;
vector<int> a(n),b(n);
vector<pair<int,int> > c,d;
vector<int>::iterator it;
for(int i=0;i<n;i++)
{
cin >>a[i]>>b[i];
if(a[i]>b[i]){x++;c.push_back({a[i],i});}
else {y++;d.push_back({b[i],i});}
}
z=c.size();
if(x>=y)
{
cout <<x<<endl;
sort(c.begin(),c.end());
for(int i=0;i<z;++i)
{
cout <<c[i].second+1<<' ';
}
}
else
{
cout <<y<<endl;
sort(d.begin(),d.end());
for(int i=n-z-1;i>=0;--i)
{
cout <<d[i].second+1<<' ';
}
}
return 0;
}