#[macro_export] macro_rules! setup { { $input:ident } => { use std::io::Read; let mut buf = String::new(); std::io::stdin().read_to_string(&mut buf).unwrap(); let mut $input = buf.split_whitespace(); }; } #[macro_export] macro_rules! parse_next { ($input:ident) => { $input.next().unwrap().parse().unwrap() }; } fn main() { setup! { input }; let s: String = parse_next!(input); let (first, last) = s.split_at(s.len() - 2); let ai_name = if last == "ai" { first.to_owned() + "AI" } else { s + "-AI" }; println!("{}", ai_name); }