C언어

[프로그래머스-C언어] 암호 해독

랏소베어123 2024. 5. 26. 19:18

문제 설명
군 전략가 머쓱이는 전쟁 중 적군이 다음과 같은 암호 체계를 사용한다는 것을 알아냈습니다.

암호화된 문자열 cipher를 주고받습니다.
그 문자열에서 code의 배수 번째 글자만 진짜 암호입니다.
문자열 cipher와 정수 code가 매개변수로 주어질 때 해독된 암호 문자열을 return하도록 solution 함수를 완성해주세요.

제한사항
1 ≤ cipher의 길이 ≤ 1,000
1 ≤ code ≤ cipher의 길이
cipher는 소문자와 공백으로만 구성되어 있습니다.
공백도 하나의 문자로 취급합니다.

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


// 파라미터로 주어지는 문자열은 const로 주어집니다. 변경하려면 문자열을 복사해서 사용하세요.
char* solution(const char* cipher, int code) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    
    int len=strlen(cipher);
    
    char* answer = (char*)malloc(len/code+1);
    int k=0;
    
    for(int i=code ;i<=len ;i+=code){
        answer[k]=cipher[i-1];
        k++;
    }
    answer[k]=NULL;
    return answer;
}

malloc으로 (len/code)로 크기할당하니 자꾸 오류가 났다.

(len/code+1)로 할당하니 오류 해결!