华为9.02笔试第一题

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

struct mydata
{
	int i;
	int cnt;
	mydata(int i, int cnt):i(i),cnt(cnt){}
};

bool operator<(const mydata& a, const mydata& b)
{
	if (a.cnt < b.cnt)
	{
		return true;
	}
	else if (a.cnt == b.cnt && a.i > b.i)
		return true;
	return false;
}

int main()
{
	int n;
	cin >> n;
	int cnt, color;
	vector<mydata> v1;
	vector<mydata> v2;

	for (int i = 1; i <= n; ++i)
	{
		cin >> cnt >> color;
		if (color == 1)
			v1.emplace_back(mydata{ i, cnt });
		else
			v2.emplace_back(mydata{ i, cnt });

	}
	sort(v1.begin(), v1.end());
	sort(v2.begin(), v2.end());
	int ans1 = 0;
	int i = 3;
	vector<int> v3;
	if (v1.size() >= 3)
	{
		while (i--)
		{
			v3.push_back((*(v1.end() - i - 1)).i);
			ans1 += (*(v1.end() - i - 1)).cnt;
		}
	}
	int ans2 = 0;
	vector<int> v4;
	if (v2.size() >= 3)
	{
		i = 3;
		while (i--)
		{
			v4.push_back((*(v2.end() - i - 1)).i);
			ans2 += (*(v2.end() - i - 1)).cnt;
		}
	}
	

	if (ans1 >= ans2&&v1.size()>=3)
	{
		sort(v3.begin(), v3.end());
		for (auto i : v3)
			cout << i<<" ";
		cout << "\n" << 1 << "\n";
		cout << ans1;
	}
	else if (ans1 <= ans2 && v2.size() >= 3)
	{
		sort(v4.begin(), v4.end());
		for (auto i : v4)
			cout << i<<" ";
		cout << "\n" << 2 << "\n";
		cout << ans2;
	}
	else
	{
		cout << "null";
	}
	return 0;
}