17413번: 단어 뒤집기 2
17413번: 단어 뒤집기 2
문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다. 먼저, 문자열 S는 아래와과 같은 규칙을 지킨다. 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져
www.acmicpc.net
단어 뒤집기 2
문제
문자열 S가 주어졌을 때, 이 문자열에서 단어만 뒤집으려고 한다.
먼저, 문자열 S는 아래와과 같은 규칙을 지킨다.
- 알파벳 소문자('a'-'z'), 숫자('0'-'9'), 공백(' '), 특수 문자('<', '>')로만 이루어져 있다.
- 문자열의 시작과 끝은 공백이 아니다.
- '<'와 '>'가 문자열에 있는 경우 번갈아가면서 등장하며, '<'이 먼저 등장한다. 또, 두 문자의 개수는 같다.
태그는 '<'로 시작해서 '>'로 끝나는 길이가 3 이상인 부분 문자열이고, '<'와 '>' 사이에는 알파벳 소문자와 공백만 있다. 단어는 알파벳 소문자와 숫자로 이루어진 부분 문자열이고, 연속하는 두 단어는 공백 하나로 구분한다. 태그는 단어가 아니며, 태그와 단어 사이에는 공백이 없다.
입력
첫째 줄에 문자열 S가 주어진다. S의 길이는 100,000 이하이다.
출력
첫째 줄에 문자열 S의 단어를 뒤집어서 출력한다.
예제 입력 1
baekjoon online judge
예제 출력 1
noojkeab enilno egduj
예제 입력 2
<ab cd>ef gh<ij kl>
예제 출력 2
<ab cd>fe hg<ij kl>
풀이
문제를 이해해보자.
- 알파벳 소문자 + 숫자 + 공백 + 특수문자("<" , ">") 로 구성
- "<"와 ">"는 태그로, 태그 안에 단어는 단어 취급 x
- 태그를 제외한 단어를 뒤집어라.
설계
문제를 다음과 같이 설계할 수 있다.
1️⃣ "<"로 시작해 ">"로 끝나면 태그이므로, 바로 출력
2️⃣ 태그가 아니면 "공백"으로 단어 구분
3️⃣ 단어 뒤집어 출력
하지만 위의 설계는 문제가 있다.
다음은 처음 설계한 코드이다.
let S = readLine()!.split(separator: ">").map{String($0)}
for word in S{
if let firstIndex = word.firstIndex(of: "<"), firstIndex == word.startIndex {
print("\(word)>", terminator: "")
}else{
if word.firstIndex(of: "<") != nil{
let temporaryWord = word.split(separator: "<").map{String($0)}
print(String(temporaryWord[0].reversed()),terminator: "")
print("<\(temporaryWord[1])>",terminator: "")
}else{
let temporaryWord = word.split(separator: " ").map{String($0)}
for w in temporaryWord{
print(String(w.reversed()), terminator: " ")
}
}
}
}
<ab cd>ef gh<ij kl>
S = ["<ab cd", "ef gh<ij kl"]
- 정답: <ab cd>fe hg<ij kl>
- 내 결과: <ab cd>hg fe<ij kl>
태그는 잘 출력이 되지만 위의 예제의 ef gh를 출력하면 hg fe가 나오는 것이다.
["ef", "gh"] 이런 식으로 공백을 기준으로 단어를 분리해야 하는데 ["ef gh"]로 받아버린 것이 문제였다.
구현
위의 문제점을 해결하기 위해 단어를 공백으로 구분하고, 구분한 단어를 뒤집고 공백을 넣어줬다.
1. ">" 찾아서 split(">")
2. 분리한 것 중 "<" 찾기
- true => split("<") 태그와 단어 분리 -> split(" ")으로 공백 구분하여 단어 뒤집기
- false => split(" ")으로 공백 구분하여 단어 뒤집기
func printReversedWords(_ words: [String]) {
for (wIndex, w) in words.enumerated() {//wIndex는 words 인덱스, w는 words
print(String(w.reversed()), terminator: "")
if wIndex < words.count - 1 {
print(" ", terminator: "")
}
}
}
전체코드
let S = readLine()!.split(separator: ">").map{String($0)}
for word in S{
if let firstIndex = word.firstIndex(of: "<"), firstIndex == word.startIndex {
print("\(word)>", terminator: "")
}else{
if word.firstIndex(of: "<") != nil{
let temporaryWord = word.split(separator: "<").map{String($0)}
let gapWord = temporaryWord[0].split(separator: " ").map{String($0)}
printReversedWords(gapWord)
print("<\(temporaryWord[1])>",terminator: "")
}else{
let gapWord = word.split(separator: " ").map{String($0)}
printReversedWords(gapWord)
}
}
}
func printReversedWords(_ words: [String]) {
for (wIndex, w) in words.enumerated() {
print(String(w.reversed()), terminator: "")
if wIndex < words.count - 1 {
print(" ", terminator: "")
}
}
}