[20291] 파일 정리

🌿 문제

[20291] 파일 정리

 

 

🌿 해결 방법

이 문제의 핵심은 파일 이름은 필요없고 오로지 '.' 뒤의 확장자만 필요하다는 것.

그래서 입력을 받을 때부터 전체 파일명을 전체 저장할 필요없이 확장자만 저장하면 된다.

string 으로 파일명을 입력받으면 확장자를 한번에 빠르게 구할 수 있는데,

먼저 find 함수를 사용해서 '.'이 몇번째 인덱스에 있는지 찾는다.

s.find(".")

그 다음 substr 함수를 사용해서 '.' 이후의 문자열을 한번에 구한다.

s.substr(s.find(".")+1)

이렇게 확장자를 구했으면 map에다가 차례대로 저장해주면 된다.

map은 key값을 기준으로 자동 정렬이 되어있는데 그걸 깜빡하고 벡터를 추가한뒤에 또 정렬을 하느라 시간, 메모리 모두 엄청나게 나왔었다ㅎㅎ..

암튼 map에 저장을 했으면 iterator을 활용해서 맵의 첫번째 원소부터 마지막 원소까지 차례대로 출력해주면 끝!

map<string, int>::iterator iter;
for(iter = m.begin(); iter != m.end(); iter++) {
    cout << iter -> first << " " << iter -> second << '\n';
}

 

 

🌿 내 코드

// [20291] 파일 정리
// 문자열, 정렬, 맵
#include <iostream>
#include <map>
using namespace std;

map<string, int> m;

void categorizeExtension(string extension)
{
    if(m.find(extension) != m.end()) {
        m[extension]++;
    }
    else {
        m.insert(make_pair(extension, 1));
    }
}

void input()
{
    int n;
    string s;
    cin >> n;
    for(int i = 0; i < n; i++) {
        cin >> s;
        categorizeExtension(s.substr(s.find(".")+1));
    }
}

void output()
{
    map<string, int>::iterator iter;
    for(iter = m.begin(); iter != m.end(); iter++) {
        cout << iter -> first << " " << iter -> second << '\n';
    }
}

int main(void)
{
    input();
    output();
}