백준 2535 아시아 정보올림피아드 (c++) : 피너클

https://www.acmicpc.net/problem/2535

2535번: 아시아 정보올림피아드

간단한 구현문제다.

cin >> n;
for (int i = 0; i < n; i++) {
	num[i + 1] = 0;
	int a, b, c;
	cin >> a >> b >> c;
	v.push_back({ c, {a, b} });
}
sort(v.begin(), v.end(), greater<pair<int, pair<int, int>>>());

벡터에 입력받는다. 받을때 {점수, {참가국, 학생번호}} 순으로 받는다. 그후 크기순으로 정렬한다.

int a = 0;
for (int i = 0; i < n; i++) {
	if (num[v[i].second.first] >= 2) continue;
	a++;
	num[v[i].second.first]++;
	cout << v[i].second.first << ' ' << v[i].second.second << endl;
	if (a == 3) break;
}

a는 현재까지 출력한 횟수를 나타낸다.

반복문을 돌리며 만약 현재 국가가 2번이상 나타난 국가라면 뒤의 문장들을 무시한다.

아니라면  a++해주고 num[현재국가]++해준다.

그후 참가국과 학생 번호를 출력하고 만약 a가 3이라면 3번 출력한것이니 break한다.

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

int n;
vector<pair<int, pair<int, int>>> v;
int num[101];

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

	cin >> n;
	for (int i = 0; i < n; i++) {
		num[i + 1] = 0;
		int a, b, c;
		cin >> a >> b >> c;
		v.push_back({ c, {a, b} });
	}
	sort(v.begin(), v.end(), greater<pair<int, pair<int, int>>>());

	int a = 0;
	for (int i = 0; i < n; i++) {
		if (num[v[i].second.first] >= 2) continue;
		a++;
		num[v[i].second.first]++;
		cout << v[i].second.first << ' ' << v[i].second.second << endl;
		if (a == 3) break;
	}
}

전체코드다.