Kangaroo

https://www.hackerrank.com/challenges/kangaroo/problem

 

두 캥거루가 같은 시간에 같은 지점에 도착하는 경우에 대해 판단하는 문제이다.

어려운 문제는 아니지만 캥거루들이 움직일 수 있는 길이의 한도가 정해져있지 않아 그 부분을 어떻게 해야하나 고민했다.

 

char* kangaroo(int x1, int v1, int x2, int v2) {
    int s1=0, s2=0;
    if((x1<x2 && v1<v2)||(x1>x2 && v1>v2))  return "NO";
    else{
        for(int i=0;;i++){
            s1= x1+v1*i;
            s2 = x2 + v2 *i;
            if(s1 == s2)    return "YES";
            if(x1<=x2){
                if(s1>s2)   return "NO";
            }
            else{
                if(s1<s2)   return "NO";
            }
        }
    }

}

△ 작성한 부분

가장 먼저 절대 만날 수 없는 경우에 대해 조건을 제시했다. >> return "NO"

그 다음 만날 수도 있고 만나지 못할 수 있는 경우에 대해 처리했다.

i 번 점프를 했을 때 각 캥거루의 위치를 나타내는 좌표값을 각각 s1, s2라고 설정했고 그 값이 같을 때는 동시에 같은 위치에 도착한 것이기 때문에 return "YES"를 한다.

그러나 그 조건을 지나치고 엇나가는 경우에 대해 따로 경우를 나눠서 return "NO"를 했다.

#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*);

// Complete the kangaroo function below.

// Please either make the string static or allocate on the heap. For example,
// static char str[] = "hello world";
// return str;
//
// OR
//
// char* str = "hello world";
// return str;
//
char* kangaroo(int x1, int v1, int x2, int v2) {
    int s1=0, s2=0;
    if((x1<x2 && v1<v2)||(x1>x2 && v1>v2))  return "NO";
    else{
        for(int i=0;;i++){
            s1= x1+v1*i;
            s2 = x2 + v2 *i;
            if(s1 == s2)    return "YES";
            if(x1<=x2){
                if(s1>s2)   return "NO";
            }
            else{
                if(s1<s2)   return "NO";
            }
        }
    }

}

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

    char** x1V1X2V2 = split_string(readline());

    char* x1_endptr;
    char* x1_str = x1V1X2V2[0];
    int x1 = strtol(x1_str, &x1_endptr, 10);

    if (x1_endptr == x1_str || *x1_endptr != '\0') { exit(EXIT_FAILURE); }

    char* v1_endptr;
    char* v1_str = x1V1X2V2[1];
    int v1 = strtol(v1_str, &v1_endptr, 10);

    if (v1_endptr == v1_str || *v1_endptr != '\0') { exit(EXIT_FAILURE); }

    char* x2_endptr;
    char* x2_str = x1V1X2V2[2];
    int x2 = strtol(x2_str, &x2_endptr, 10);

    if (x2_endptr == x2_str || *x2_endptr != '\0') { exit(EXIT_FAILURE); }

    char* v2_endptr;
    char* v2_str = x1V1X2V2[3];
    int v2 = strtol(v2_str, &v2_endptr, 10);

    if (v2_endptr == v2_str || *v2_endptr != '\0') { exit(EXIT_FAILURE); }

    char* result = kangaroo(x1, v1, x2, v2);

    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;
}