Cats and a Mouse

고양이 A와 B, 쥐 C사이의 거리를 이용해 경우에 따라 서로 다른 문자열을 출력하도록 하는 함수 작성

www.hackerrank.com/challenges/cats-and-a-mouse/problem

 

char* catAndMouse(int x, int y, int z) {
    int d1 = abs(x-z);
    int d2 = abs(y-z);
    if(d1 > d2) return "Cat B";
    else if(d1 < d2)    return "Cat A";
    else {
    return "Mouse C";
    }

}

△ 작성한 부분

abs 함수를 사용해 x와 z, y와 z의 거리를 절대값으로 각각 d1, d2 변수에 저장한다.

그리고 그 값을 비교해 필요한 문자열을 출력한다.

 

#include <assert.h>
#include <limits.h>
#include <math.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

char* readline();
char** split_string(char*);

char* catAndMouse(int x, int y, int z) {
    int d1 = abs(x-z);
    int d2 = abs(y-z);
    if(d1 > d2) return "Cat B";
    else if(d1 < d2)    return "Cat A";
    else {
    return "Mouse C";
    }

}

int main()
{
    FILE* fptr = fopen(getenv("OUTPUT_PATH"), "w");

    char* q_endptr;
    char* q_str = readline();
    int q = strtol(q_str, &q_endptr, 10);

    if (q_endptr == q_str || *q_endptr != '\0') { exit(EXIT_FAILURE); }

    for (int q_itr = 0; q_itr < q; q_itr++) {
        char** xyz = split_string(readline());

        char* x_endptr;
        char* x_str = xyz[0];
        int x = strtol(x_str, &x_endptr, 10);

        if (x_endptr == x_str || *x_endptr != '\0') { exit(EXIT_FAILURE); }

        char* y_endptr;
        char* y_str = xyz[1];
        int y = strtol(y_str, &y_endptr, 10);

        if (y_endptr == y_str || *y_endptr != '\0') { exit(EXIT_FAILURE); }

        char* z_endptr;
        char* z_str = xyz[2];
        int z = strtol(z_str, &z_endptr, 10);

        if (z_endptr == z_str || *z_endptr != '\0') { exit(EXIT_FAILURE); }

        char* result = catAndMouse(x, y, z);

        fprintf(fptr, "%s\n", result);
    }

    fclose(fptr);

    return 0;
}

char* readline() {
    size_t alloc_length = 1024;
    size_t data_length = 0;
    char* data = malloc(alloc_length);

    while (true) {
        char* cursor = data + data_length;
        char* line = fgets(cursor, alloc_length - data_length, stdin);

        if (!line) { break; }

        data_length += strlen(cursor);

        if (data_length < alloc_length - 1 || data[data_length - 1] == '\n') { break; }

        size_t new_length = alloc_length << 1;
        data = realloc(data, new_length);

        if (!data) { break; }

        alloc_length = new_length;
    }

    if (data[data_length - 1] == '\n') {
        data[data_length - 1] = '\0';
    }

    data = realloc(data, data_length);

    return data;
}

char** split_string(char* str) {
    char** splits = NULL;
    char* token = strtok(str, " ");

    int spaces = 0;

    while (token) {
        splits = realloc(splits, sizeof(char*) * ++spaces);
        if (!splits) {
            return splits;
        }

        splits[spaces - 1] = token;

        token = strtok(NULL, " ");
    }

    return splits;
}

 △ 전체코드