[Swift] 입력 받기 - readLine() 처리 방법 살펴보기 (feat.백준 알고리즘 풀 때 필수)

안녕하세요 감자입니다! 오늘은 swift 언어에서 입력을 받는 메서드인 readLine()에 대해서 알아보거에요. 코딩테스트를 준비중인데 프로그래머스는 답안을 func로 작성하며 되지만, 백준은 입력과 출력을 모두 입력해야해요! 그래서 백준으로 공부할 때 각 언어별로 input을 받는 방법에 대해 익히고 있는것이 필수인데요, 

입력을 그냥 수 하나만 받는게 아니라, 행렬로 받기도 하고, 배열로 받기도하고, 여러줄로 받기도하고 정말 다양한 방법으로 input을 받잖아요! 이것을 모두 익혀놔야 나중에 알고리즘 풀때 막히지 않기 때문에 차근차근 정리해보도록 할게요.

사실 제가 헷갈리고 제가 외워놓기(?) 위해서 적는거랍니다 허허 그럼 go

 

readLine()

Swift에서 입력은 readLine()을 통해서 할 수 있다. readLine()은 입력을 모두 optional String형으로 처리하기 때문에 또다른 처리가 필요하다. 또한 Swfit에서 readLine은 EOF가 들어오기전의 입력을 return 하게 되는데, EOF란 End Of File 의 뜻으로 엔터가 입력되는 것을 EOF처기가 됐다~ 라고 말한다. 즉, 엔터가 한번 입력되면 readLine() 한개가 끝나는 것이다. 

따라서 Swift에서는 입력을 여러줄 받을 때 readLine()을 여러번 써주면 되는 것이다! 

1. 정수 입력받기

앞서 readLine은 모든 값을 optional string의 형태로 return한다고 했다. 따라서 추가적인 처리를 해줘야, 정수 입력이 가능하다. 강제로 옵셔널을 벗기는 기능과 , INT로 형변환이 필요하다.

let input = readLine()
print(type(of:input))
// prints "Optional<String>"

let inputOut = input!
print(type(of:inputOut))
//prints "String"

let intvar = Int(InputOut)
print(type(of:intvar))
//prints "Optional<Int>"

let result = intvar!
print(type(of:result))
//prints "Int"

위 코드를 보면, readLine으로 입력받은 input은 optional string 값이다. (string? 값)

그래서 이를 inputOut의 변수에 언래핑하여 저장하였더니, 일반 String형이 출력된다. 그럼 다시 이를 숫자로 바꾸고 싶다면 어떻게 해야할까.

intvar 변수에 Int() 를 활용하여 형변환을 시도했다. 하지만, 다시 Optional Int 값으로 출력되었고, 옵셔널이 언래핑 되지 않았다. 왜 자동으로 다시 옵셔널로 래핑되는 것일까? 그 이유는, string으로 입력된 1,2,3,4 와 같은 숫자들은 int로 형변환이 가능하지만, 만약에 입력받은 string이 숫자가 아닌 a,b,c,d 와 같은 문자라면, int로 형변환을 시도하면 nil이 되기 때문이다. 그래서 안전한 언어인 Swift는 이를 다시 래핑하여 optional로 출력하는 것이다. 따라서 정수로 출력하기 위해서는 마지막에 다시한번 강제로 언래핑 해주어야 한다. (result를 봐라. 다시 언래핑 해주니까 우리가 원했던 정수형이 드디어 출력되었다.)

이를 한번에 입력하고 싶다면 아래와 같이 코드를 작성하면 된다.

let n = Int(readLine()!)!

 

2. 문자열 하나 입력 받기

앞에서 정수를 입력하기 위해서는 옵셔널을 두번 벗겨주어야 했다. 하지만 이미 string 형태로 입력을 받기 때문에 문자열 하나를 입력받기 위해서는 readLine에 옵셔널을 한번만 강제로 벗겨주면 된다.

let text = readLine()!

 

3. 띄어쓰기로 구분하여 입력받기, 바로 배열로 입력받기

import Foundation


let input = readLine()!  // h e l l o 라고 입력
print(type(of: input))
print(input)
// prints "String"
// prints "h e l l o"

let result = input.components(separatedBy: " ")
print(type(of: result))
print(result)
// prints "Array<String>"
// prints ["h", "e", "l", "l", "o"]

fountdation 프레임워크에 포함되어있는 components 메서드를 활용하여 띄어쓰기로 구분해주었다. 이렇게 쓰게 되면 띄어쓰기를 기준으로 입력받은 문자열이 "배열" 로 바로 return된다. 유용할것 같다!

 

4. 띄어쓰기로 정수 여러개 입력 받기

정수를 여러개 입력받기 위해서는, split 과 map을 이용해 줄것이다. 코드는 아래와 같다.

let input = readLine()!.split(separator: " ").map { Int(String($0))! }
// 입력: 1 2 3
// Array<Int>
// [1, 2, 3]

map() 메서드를 클로저로 사용한 케이스이다. 배열 하나하나의 아이템에 map 클로저 안의 코드를 적용시킨다는 뜻으로, 하나하나 Int로 바꿔주어 배열에 매핑시켜준다고 이해하면 편할 것이다.

 

5. (띄어쓰기없이) 한 번에 입력 받아서 배열로 return하기

let input = Array(readLine()!) //입력; hello
print(input)
print(type(of:input))

//prints ["h", "e", "l", "l", "o"]
// prints Array<String>

흔하진 않지만, 간혹 띄어쓰기 없이 입력을 받는 경우가 있다. 그럴 때 위와 같은 방법을 사용하면 좋다. 알아서 저절로 한글자씩 배열의 한 인덱스에 저장해준다. 편한것 같다. 이를 int형으로 입력받고 싶다면 아래와 같이 작성하자.

let result = input.map { Int(String($0))! }
print(result)
print(type(of: result))

//prints [1,2,3,4,5]  만약, 입력이 "12345" 였다면 이렇게 출력,
//prints Array<Int>