[백준 2456] 나는 학급회장이다 (C++)

풀이

지문이 엄청 길지만 확실하게 정리하면 다음과 같다.

전체 점수, 3점 투표수, 2점 투표수가 모두 같으면 회장을 결정할 수 없다.

전체 점수, 3점 투표수가 같으면 2점 투표수가 많은 학생이 회장이 된다.

전체 점수가 같으면 3점 투표수가 많은 학생이 회장이 된다.

 

이렇게 정렬 기준을 잡고 정렬한 다음에 첫 번째 학생과 두 번째 학생을 비교하면 해결할 수 있다.

비교하는 이유는 전체 점수, 3점 투표수, 2점 투표수가 모두 같은 경우가 존재할 수 있기 때문에

한 번 더 확인해주고 판단하기 위함이다.

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

class Student
{
public:
	int idx;
	int total;
	int point3;
	int point2;

	Student() : idx(0), total(0), point3(0), point2(0) {}
	Student(int idx, int a, int b, int c) : idx(idx), total(a), point3(b), point2(c) {}
	bool operator<(const Student& other)
	{
		if (total == other.total)
		{
			if (point3 == other.point3)
			{
				return point2 > other.point2;
			}
			return point3 > other.point3;
		}
		return total > other.total;
	}

	bool operator==(const Student& other)
	{
		if (total == other.total && point3 == other.point3 && point2 == other.point2)
			return true;
		else
			return false;
	}
};

void addPoint(vector<Student>& v, int idx, int point)
{
	if (point == 3)
		v[idx].point3++;
	else if (point == 2)
		v[idx].point2++;

	v[idx].total += point;
}

int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);

	int n;
	cin >> n;
	vector<Student> v(3);

	for (int i = 0; i < 3; i++)
		v[i].idx = i + 1;

	for (int i = 0; i < n; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;

		addPoint(v, 0, a);
		addPoint(v, 1, b);
		addPoint(v, 2, c);
	}

	sort(v.begin(), v.end());

	if (v[0] == v[1])
		cout << 0 << ' ' << v[0].total << '\n';
	else
		cout << v[0].idx << ' ' << v[0].total << '\n';
	return 0;
}