末尾の文字 (Last Letter)

문제 출처 : https://www.acmicpc.net/problem/27541

 

언어 : Kotlin

 

문제 설명 :

JOI 高校の生徒である葵は，文字列 JOIG が，文字列 JOI の末尾に文字 G を付け加えてできることに気が付いた．

そこから葵は，様々な文字列について，末尾に文字 G を付け加えたり，末尾の文字 G を取り除いて遊ぶようになった．

具体的には，葵は文字列を見ると次のように遊ぶ．

見た文字列の末尾の文字が G のとき，末尾の文字 G を取り除いた文字列を思い浮かべる．取り除く文字は末尾の 1 文字のみである．
見た文字列の末尾の文字が G でないとき，文字列の末尾に文字 G を付け加えた文字列を思い浮かべる．
長さ N の文字列 S が与えられる．葵が文字列 S を見たとき思い浮かべる文字列を求めるプログラムを作成せよ．

 

입력 :

入力は以下の形式で与えられる．

N
S

 

출력 :

葵が文字列 S を見たとき思い浮かべる文字列を出力せよ．

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 1024MB
• 2 ≦ N ≦ 100．
• S は長さ N の文字列である．
• S の各文字は英大文字である．
• N は整数である．

 

입출력 예 :

입력	출력
4 JOIG	JOI
3 JOI	JOIG
3 EGG	EG

 

풀이 : 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val n = readLine().toInt()
    val s = readLine()

    bw.write(if (s.last() == 'G') s.removeSuffix("G") else "${s}G")
    bw.flush()
    bw.close()
}