취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 25325번 문제인 학생 인기도 측정이다.
문제는 아래 링크를 확인하자.

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

각 학생별로 자신을 좋아하는 학생의 수를 세고, 주어진 비교 기준에 따라 학생들을 정렬해 문제를 해결하자.

한 학생이 좋아하는 학생 리스트에는 중복된 학생이 등장하지 않으므로, 그냥 입력이 끝날 때까지 들어오는 모든 이름에 대하여 해당 이름을 가진 학생을 좋아하는 사람을 한 명씩 증가시키는 구현을 해도 문제를 충분히 해결할 수 있다.

각 학생들에게 입력순서대로 정수 인덱스를 부여해 구현하면 구현을 편하게 할 수 있다. 이는 map을 이용해 구현 가능하다.

첫 정렬 기준이 좋아하는 학생 수 "내림차순", 두번째 정렬 기준이 이름 "오름차순"임에 유의하자. 이를 쉽게 구현하기 위한 트릭으로 좋아하는 학생 수가 K일 때 K 대신 -K를 저장하는 방법이 있다. 이 경우 두 기준 모두 "오름차순"으로 정렬하면 되므로 std::sort와 pair을 이용해 구현을 간단히 할 수 있게 된다.

아래는 제출한 소스코드이다.

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

int N;
map<string, int> strtoidx;
pair<int, string> pr[100];

string s;

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

	cin >> N;
	for (int idx = 0; idx < N; idx++) {
		cin >> s;
		strtoidx.insert(make_pair(s, idx));
		pr[idx].second = s;
	}

	while (cin >> s) pr[strtoidx[s]].first--;
	
	sort(pr, pr + N);

	for (int i = 0; i < N; i++) cout << pr[i].second << ' ' << -pr[i].first << '\n';
}