[C/C++] 백준 20291번 - 파일 정리

문제

친구로부터 노트북을 중고로 산 스브러스는 노트북을 켜자마자 경악할 수밖에 없었다. 바탕화면에 온갖 파일들이 정리도 안 된 채 가득했기 때문이다. 그리고 화면의 구석에서 친구의 메시지를 확인할 수 있었다.

바탕화면의 파일들에는 값진 보물에 대한 정보가 들어 있어. 하나라도 지우게 된다면 보물은 물론이고 다시는 노트북을 쓸 수 없게 될 거야. 파일들을 잘 분석해서 보물의 주인공이 될 수 있길 바랄게. 힌트는 “확장자”야.

화가 났던 스브러스는 보물 이야기에 금세 화가 풀렸고 보물의 정보를 알아내려고 애썼다. 하지만 파일이 너무 많은 탓에 이내 포기했고 보물의 절반을 보상으로 파일의 정리를 요청해왔다. 스브러스의 요청은 다음과 같다.

• 파일을 확장자 별로 정리해서 몇 개씩 있는지 알려줘
• 보기 편하게 확장자들을 사전 순으로 정렬해 줘

그럼 보물의 절반을 얻어내기 위해 얼른 스브러스의 노트북 파일 정리를 해줄 프로그램을 만들자!

입력

첫째 줄에 바탕화면에 있는 파일의 개수 N이 주어진다. (1≤N≤50 000)

둘째 줄부터 N개 줄에 바탕화면에 있는 파일의 이름이 주어진다. 파일의 이름은 알파벳 소문자와 점(.)으로만 구성되어 있다. 점은 정확히 한 번 등장하며, 파일 이름의 첫 글자 또는 마지막 글자로 오지 않는다. 각 파일의 이름의 길이는 최소 3, 최대 100이다.

출력

확장자의 이름과 그 확장자 파일의 개수를 한 줄에 하나씩 출력한다. 확장자가 여러 개 있는 경우 확장자 이름의 사전순으로 출력한다.

 

<코드>

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

int N, cnt = 0;
vector<pair<string, int>> ans;
vector<string> ex; // extension
string s, tmp;

int main()
{
	cin >> N;

	// 뒤에 확장자만 남겨서 ex벡터에 담는 과정
	for (int i = 0; i < N; i++)
	{
		cin >> s;
		s.erase(0, s.find('.') + 1);
		ex.push_back(s);
		
	}
	sort(ex.begin(), ex.end()); // 벡터 사전순으로 정렬

	tmp = ex[0]; // 초기값
	for (int i = 0; i < ex.size(); i++)
	{
		if (tmp == ex[i]) cnt++;  // 같은 확장자 개수 세기
		else if(tmp != ex[i]) // 새로운 확장자등장! 
		{
			// 확장자명이랑 이때까지 개수 센거 ans벡터에 담기 
			ans.push_back(pair<string,int>(tmp, cnt));
			// 다시 리셋
			tmp = ex[i];
			cnt = 1;
		}
	}
	// 맨 마지막에 확장자명이랑 개수 안담긴거 마저 담기
	ans.push_back(pair<string, int>(tmp, cnt));

	for (int i = 0; i < ans.size(); i++)
	{
		cout << ans[i].first << " " << ans[i].second << endl;
	}
}

 

풀이 방법

erase함수를 이용해 입력된 문자열에서 확장자만 빼고 모두 지운뒤 확장자들만 모아놓는 ex벡터에 push 했습니다. 그런 뒤 정렬을 하게 되면 아래와 같이 사전 순으로 정렬되게 됩니다.

txt	->	icpc
spc		icpc
icpc		spc
icpc		spc
txt		txt
world		txt
spc		txt
txt		world

 

그리고 ex벡터를 처음부터 마지막까지 탐색하면서 같은 것의 개수를 세고 같은 종류를 다 세었다면 해당 확장자와 개수를 ans 벡터에 push하여 풀이를 하였습니다. 따라서 O(N) 시간복잡도만에 문제를 해결할 수 있습니다.

 

 

www.acmicpc.net/problem/20291

20291번: 파일 정리