백준 17413번 Rust 풀이

러스트를 연습하려고 풀어본 문제인데 풀고보니 match 표현식을 활용해봤으면 더 좋았을거 같습니다.

문제 링크
https://www.acmicpc.net/problem/17413

17413번: 단어 뒤집기 2

 

풀이

use std::io::{self};
 
fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input);
 
    // <를 찾으면 true
    let mut is_angle = false;
    let mut tmp_word = String::new();
    let mut answer = String::new();
 
    for ch in input.chars() {
        if ch == '<' {
            is_angle = true;
            if tmp_word.len() == 0 {
                answer.push(ch);
                continue;
            } else {
                let converted = reverse_word(&mut tmp_word);
                answer.push_str(&converted);
                answer.push('<');
                tmp_word = String::new();
                continue;
            }
 
        }
        if ch == '>' {
            answer.push(ch);
            is_angle = false;
            continue;
        }
        // 꺽쇠 안에 있는 단어들은 뒤집지 않으므로
        if is_angle {
            answer.push(ch);
            continue;
        }
        // rust stdin 특성상 마지막은 줄바꿈 문자임
        if ch == ' ' || ch == '\n' {
            if tmp_word.len() > 0 {
                let converted = reverse_word(&mut tmp_word);
                answer.push_str(&converted);
                answer.push(' ');
                tmp_word = String::new();
            } else { // 공백이 이어지는 경우
                answer.push(' ');
            }
        } else {
            tmp_word.push(ch);
        }
    }
    println!("{}", answer);
 
}
 
fn reverse_word(word: &mut String) -> String {
    let word_byte = word.as_bytes();
    let mut converted = String::new();
    for i in (0..word.len()).rev() {
        converted.push(word_byte[i] as char);
    }
 
    converted
}