비밀편지 (feat. 백준, 2596번)

비밀편지
https://www.acmicpc.net/problem/2596

2596번: 비밀편지

 

// woohyeon.kim
// kim519620.tistory.com
#include <iostream>
#include <string>
using namespace std;
int main() {
	ios::sync_with_stdio(false), cin.tie(NULL);
    string arr[] = {"000000", "001111", "010011", "011100", "100110", "101001", "110101", "111010"};
    string tmp, ans;
    register int N;
    cin >> N >> tmp;
    for(register int n = 0; n < N; ++n){
        auto str = tmp.substr(n * 6, 6);
        register int idx = -1;
        for(register int i = 0; i < 8; ++i)
            if(!str.compare(arr[i])){
                idx = i;
                break;
            }
        if(idx != -1)
            ans += (char)('A' + idx);
        else{
            for(register int i = 0; i < 8; ++i){
                register int cnt = 0;
                for(register int j = 0; j < 6; ++j)
                    if(str.at(j) != arr[i].at(j))
                        ++cnt;
                if(cnt == 1){
                    ans += (char)('A' + i);
                    idx = 0;
                    break;
                }
            }
            if(idx != 0){
                cout << n + 1, ans.clear();
                break;
            }
        }
    }
    if(!ans.empty())
        cout << ans;
    return 0;
}
// *&)*@*

 

반응형

문제의 조건처럼 문자열의 문자를 검사하여 출력하는 문제입니다.