[Algorithm] 백준 1157 단어 공부 (Swift/C++)

1157번: 단어 공부

알고리즘 분류

• 구현
• 문자열

문제 풀이

먼저 문자를 입력 받고, 알파벳의 숫자를 셀 배열 alphabet을 생성한다. alphabet 배열들의 요소는 모두 0으로 초기화해준다.

 

문제에서 가장 많이 사용된 문자를 대문자로 출력하라고 했기 때문에, for 문을 돌면서, word의 문자 중 소문자인 것이 있다면 32를 빼서 대문자로 바꿔준다.

alphabet 배열의 0번째 인덱스는 A가 나온 횟수, 1번째 인덱스는 B가 나온 횟수, .. 25번째 인덱스는 Z가 나온 횟수가 된다.

따라서 아스키코드값에서 65를 뺀 값을 alphabet 배열의 인덱스로 해서 1씩 증가시켜준다.

 

alphabet 배열의 0번째 인덱스부터 마지막 인덱스까지 for 문을 돌면서 제일 큰 값을 찾아준다.

초기값을 ans = 0으로 놓고, for 문을 돌면서 큰 값을 갱신시켜주면 된다.

만약 제일 큰 값이 2개 이상인 경우, b 값을 true로 바꿔준다.

제일 큰 값이 하나라면 b 값은 false이고 index 값에 해당 인덱스를 저장한다.

 

b가 true이면 가장 많이 사용된 알파벳이 여러 개 존재하는 것이므로 ?를,

b가 false이면 저장된 index에 65를 더해서 대문자를 출력한다.

index + 65는 Int 타입이기 때문에, UnicodeScalar를 이용하여 Type Conversion을 해서 출력하면 된다.

UnicodeScalar는 Unicode.Scalar의 typealias인데, scalar 값의 Unicode를 의미한다.

아스키코드값을 문자로 바꿔주는 역할을 한다.

 

소스 코드 (Swift)

import Foundation

let word = readLine()!

var alphabet = [Int](repeating: 0, count: 26)

for i in word {
    var asciiCode = Int(i.asciiValue!)

    if asciiCode >= 97 && asciiCode <= 122 { // 소문자일 경우, 대문자로 바꿔주기.
        asciiCode -= 32
    }

    alphabet[asciiCode - 65] += 1
}

var b = false
var ans = 0
var index = 0

for i in 0...alphabet.count - 1 {
    if ans < alphabet[i] {
        b = false
        ans = alphabet[i]
        index = i
    } else if ans == alphabet[i] {
        b = true // 중복있음!
        ans = alphabet[i]
    }
}

b ?? print("?") : print(UnicodeScalar(index + 65)!) // 아스키코드값을 이용해서 대문자 출력

 

소스 코드 (C++)

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

using namespace std;

char alphabet[] = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'r', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'};
int check[26] = {0, };
int ans = 0;
int idx = 0;
string word;

int main(void) {
    cin.tie(0);
    cout.tie(0);
    std::ios::sync_with_stdio(false);
    
    cin >> word;
    
    for (int i = 0; i < word.length(); i++) {
        if (word[i] >= 'a' && word[i] <= 'z') {
            // 소문자를 대문자로 변환
            word[i] -= 32;
        }
    }
    
    for (int i = 0; i < word.length(); i++) {
        for (int j = 0; j < 26; j++) {
            // A~Z까지의 알파벳을 입력된 단어의 알파벳을 각각 비교
            if (word[i] != alphabet[j]) continue;
            else {
                // 입력된 단어의 알파벳이 알파벳 배열의 값과 일치한다면
                int num = word[i]; // 그 알파벳의 아스키 코드값을 num에 저장한다.
                // check 배열에서, check[0]에는 A, check[1]에는 B, check[2]에는 C에 해당하는 알파벳이 나온 횟수를 +1 해준다.
                // 이렇게 차례대로 저장하기 위해 check[num - 'A']하여 저장해준다.
                check[num - 'A']++;
            }
        }
    }
    
    bool b = false; // 가장 많이 사용된 알파벳이 여러 개 존재하는 경우를 체크
    for (int i = 0; i < 26; i++) {
        if (ans < check[i]) {
            // 기존에 존재하던 최고 빈도수 값 < 비교하려는 값, ans 갱신해주기
            b = false;
            idx = i;
            ans = check[i];
        }
        else if (ans == check[i]) {
            // 기존에 존재하던 최고 빈도수 값 == 비교하려는 값, ans는 갱신하지 않고, b = true로 바꿔준다.
            idx = i;
            b = true;
        }
    }
    
    if (b) cout << "?" << "\\n";
    else cout << alphabet[idx] << "\\n";
    
    return 0;
}