[백준] 1316, 그룹 단어 체커

YUN·2026년 2월 22일

C++

목록 보기
42/85

그룹 단어의 개수를 출력하는 문제이다.

핵심은 문자열을 순회하며, 직전 문자와 동일하지 않은데 이미 과거에 등장했던 문자가 등장하는 경우 를 찾아내는 것이다.

1. 나의 풀이

#include <iostream>
#include <string>
#include <algorithm>

int arr[26];    

using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int N;
    string s;
    cin >> N;
    int cnt=N;
    for(int i=0; i<N; i++) {
        int prev_idx=-1;
        cin >> s;
        for(char c:s) {
            int idx = c-'a';
            if(arr[idx]==0 || idx==prev_idx) {
                arr[idx]=1;
            } else if (arr[idx]) {
                cnt--;
                break;
            }
            prev_idx=idx;
        }
        fill(arr+0, arr+26, 0);
    }
    cout << cnt;
    
    return 0;
}

이게 처음 나의 풀이이다.

(사실 진짜 처음 풀이에서는 매 단어마다 배열을 0으로 초기화하는것을 깜빡해서 틀렸었다)

뭔가 비효율적인 면이 존재한다.

for(char : s)로 쓰다보니 직전 인덱스 저장을 위해서 추가로 prev_idx 변수를 하나 더 뒀어야했고, 배열을 매번 fill()로 초기화 해주는 비효율적인 코드가 존재한다.

다른 사람의 코드를 참고해서 이를 개선해보았다.

#include <iostream>
#include <string>

using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int N;
    string s;
    cin >> N;
    int cnt=N;
    while(N--) {
        cin >> s;
        int arr[26]={};
        arr[s[0]-'a']++;
        for(int i=1; i<s.size(); i++) {
            int idx = s[i]-'a';
            if(arr[idx]==0 || s[i]==s[i-1]) {
                arr[idx]=1;
            } else if (arr[idx]) {
                cnt--;
                break;
            }
        }
    }
    cout << cnt;
    
    return 0;
}

뭔가 큰 차이는 없는 것 같지만..비효율적인 부분들이 확실히 줄어든 것 같다.

(alogrithm 헤더 include도 없고 fill도 안쓰고 prev_idx도 없고..등등?)

2. 느낀 점

(1) 인덱스 정보가 필요한 경우 for(char c: s) 보다는 for(int i=0;i<s.size();i++)을 쓰자

인덱스 정보가 필요한 경우는 코드의 효율성 측면에서 for(int i=0;i<s.size();i++)를 쓰는게 유리한 경우가 많은 것 같다.

profile
안녕하세요. 전자공학부 학부생의 공부 기록입니다.

0개의 댓글