배열 회전시키기 코틀린(Kotlin)

오늘은 코딩테스트 입문 단계 중 그나마 난이도가 있어보이는 문제를 골라서 풀어보았다.

 

문제

문제

문제 설명만 읽어봐서는 배열을 회전시킨다는 의미를 이해하지 못했는데

direction과 입출력 예시를 보고 의미를 파악했다.

 

direction이 left이면 배열을 그대로 한 칸 왼쪽으로 밀면 되고, right이면 오른쪽으로 밀면 되는 것이었다.

 

풀이

풀이

가장 먼저 리턴될 배열의 크기를 정해주었다.

 

다른 문제를 풀 때도 배열의 크기를 정하는 게 은근 귀찮은 부분이었는데  

위의 입출력 예를 보니 direction이 right이면 리턴될 배열의 첫 칸이 주어진 배열의 마지막 원소였다.

 

반대로 direction이 left일 때, 리턴될 배열의 마지막 칸이 주어진 배열의 첫 원소였다.

 

그래서 각 상황마다 처음 혹은 마지막 원소 값을 넣어주고 나머지는 for문을 사용해 배열에 값을 대입해주었다.

 

실행 결과

실행 결과