[C++/프로그래머스]코딩테스트 세 개의 구분자

문제 설명

임의의 문자열이 주어졌을 때 문자 "a", "b", "c"를 구분자로 사용해 문자열을 나누고자 합니다.

예를 들어 주어진 문자열이 "baconlettucetomato"라면 나눠진 문자열 목록은 ["onlettu", "etom", "to"] 가 됩니다.

문자열 myStr이 주어졌을 때 위 예시와 같이 "a", "b", "c"를 사용해 나눠진 문자열을 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

단, 두 구분자 사이에 다른 문자가 없을 경우에는 아무것도 저장하지 않으며, return할 배열이 빈 배열이라면 ["EMPTY"]를 return 합니다.

제한사항

1 ≤ myStr의 길이 ≤ 1,000,000
- myStr은 알파벳 소문자로 이루어진 문자열 입니다.

입출력 예

myStr	result
"baconlettucetomato"	["onlettu", "etom", "to"]
"abcd"	["d"]
cabab"	["EMPTY"]

입출력 예 설명

입출력 예 #1

문제 설명의 예시와 같습니다.

입출력 예 #2

"c" 이전에는 "a", "b", "c" 이외의 문자가 없습니다.
"c" 이후에 문자열 "d"가 있으므로 "d"를 저장합니다.
따라서 ["d"]를 return 합니다.

입출력 예 #3

"a", "b", "c" 이외의 문자가 존재하지 않습니다. 따라서 저장할 문자열이 없습니다.
따라서 ["EMPTY"]를 return 합니다.

C++ 코딩테스트 세 개의 구분자

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <string.h>

char** solution(const char* myStr) {
    int n = strlen(myStr);
    char** answer = (char**)malloc(sizeof(char*) * (n + 1));
    char* temp = (char*)malloc(sizeof(char) * (n + 1));
    int answerIndex = 0, tempIndex = 0;

    for(int i = 0; i < n; i++) {
        if(myStr[i] == 'a' || myStr[i] == 'b' || myStr[i] == 'c') {
            if(tempIndex > 0) {
                temp[tempIndex] = '\0';
                answer[answerIndex] = (char*)malloc(sizeof(char) * (tempIndex + 1));
                strcpy(answer[answerIndex], temp);
                answerIndex++;
                tempIndex = 0;
            }
        } else {
            temp[tempIndex++] = myStr[i];
        }
    }
    
    if(tempIndex > 0) {
        temp[tempIndex] = '\0';
        answer[answerIndex] = (char*)malloc(sizeof(char) * (tempIndex + 1));
        strcpy(answer[answerIndex], temp);
        answerIndex++;
    }

    free(temp);
    
    if(answerIndex == 0) {
        answer[0] = (char*)malloc(sizeof(char) * 6);
        strcpy(answer[0], "EMPTY");
        answerIndex++;
    }

    answer[answerIndex] = NULL; // End of array

    return answer;
}

저작자표시 비영리 변경금지

wyatt