[백준] 14426: 접두사 찾기
https://www.acmicpc.net/problem/14426
Github Solution Link
문자열의 길이도 500자고, 입력도 10000개가 들어온다고 해서 시간을 단축시키는 것에 대해서 고민을 했던 문제였다.
생각을 해보다가, 접두사라면 해당 단어와 첫 번째 글자는 무조건 같기 때문에 이 점을 이용해서 multimap을 사용하기로 했다.
- multimap : <key, value>로 구성되어 있다. 일반 map은 key값이 중복되면 안되지만, multimap은 중복되어도 된다.
이 문제처럼 단어의 첫 글자를 key로 하는 경우, 이는 충분히 중복될 수 있는 값이기 때문에 multimap을 사용하기로 했다.
multimap의 equal_range()
함수를 이용해서, 같은 key값을 가지는 값(맨 앞 자리가 같은 단어)들 내에서만 접두사가 올바른 지 체크를 진행하였다. find()
함수의 리턴값은 해당 값을 찾았을 때 시작 인덱스를 반환하는데, 그것이 0(단어의 맨 앞)이어야 올바른 접두사라고 판단할 수 있다.
처음 작성한 코드에 이 곳에 오류가 있었는데 맞았다고 해서 테스트케이스 추가 요청을 했다...!
정답 코드
#include <iostream>
#include <string>
#include <map>
#include <utility>
using namespace std;
multimap<char, string> mmap;
void fastio(void);
int main(void)
{
string s;
int n, m, ret = 0;
multimap<char, string>::iterator iter;
fastio();
cin >> n >> m;
for (int i = 0; i < n; i++)
{
cin >> s;
mmap.insert({s[0], s});
}
for (int i = 0; i < m; i++)
{
cin >> s;
for (iter = mmap.equal_range(s[0]).first; iter != mmap.equal_range(s[0]).second; iter++)
if (iter->second.find(s) == 0)
{ ret++; break;}
}
cout << ret << "\n";
return (0);
}
void fastio(void)
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
}
Author And Source
이 문제에 관하여([백준] 14426: 접두사 찾기), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@besyia0k0/백준-14426저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)