[코딩테스트 Lv.0] 문자열의 뒤의 n글자

문제 설명

문자열 my_string과 정수 n이 매개변수로 주어질 때, my_string의 뒤의 n글자로 이루어진 문자열을 return 하는 solution 함수를 작성해 주세요.

제한사항

my_string은 숫자와 알파벳으로 이루어져 있습니다.
1 ≤ my_string의 길이 ≤ 1,000
1 ≤ n ≤ my_string의 길이

입출력 예

my_string		n	result
"ProgrammerS123"	11	"grammerS123"
"He110W0r1d"		5	"W0r1d"

입출력 설명

입출력 예 #1
예제 1번의 my_string에서 뒤의 11글자는 "grammerS123"이므로 이 문자열을 return 합니다.

입출력 예 #2
예제 2번의 my_string에서 뒤의 5글자는 "W0r1d"이므로 이 문자열을 return 합니다.

---

내 풀이

Java

class Solution {
    public String solution(String my_string, int n) {
        String answer = my_string.substring(my_string.length() - n);
        return answer;
    }
}

C

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

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

Python

def solution(my_string, n):
    answer = my_string[len(my_string) - n:]
    return answer

---

https://school.programmers.co.kr/learn/courses/30/lessons/181910

프로그래머스