Monistus

use std::{char, io::*};

const MAX_CAPACITY: usize = 50000;

#[derive(Copy, Clone, Debug)]
enum Char {
    Letter(char),
    Digit(u8),
    Unused,
}

fn print_buffer(buffer: &[Char]) {
    for c in buffer {
        match c {
            Char::Digit(digit) => {
                print!("\x1b[93m{digit}\x1b[0m")
            }
            Char::Letter(l) => {
                print!("{l}")
            }
            Char::Unused => {}
        }
    }
    println!("");
}

fn main() -> Result<()> {
    let mut input = String::new();
    stdin().read_line(&mut input)?;

    let mapped_input = input
        .chars()
        .map(|char| {
            if let Ok(digit) = char.to_string().parse::<u8>() {
                Char::Digit(digit)
            } else {
                Char::Letter(char)
            }
        })
        .collect::<Vec<_>>();

    let mut buffer = [Char::Unused; MAX_CAPACITY];
    buffer[..input.len()].copy_from_slice(mapped_input.as_slice());

    let mut i = 0;
    let mut last = input.len();

    loop {
        match buffer[i] {
            Char::Digit(digit) => {
                let digit = digit as usize;
                println!("\nIndex: {}", i);
                println!("Copies:");
                if i + 1 + digit < last {
                    buffer.copy_within((i + 1 + digit)..last, i + 2 * digit);
                }
                print_buffer(&buffer);

                buffer.copy_within((i + 1)..(i + 1 + digit), i + digit);
                print_buffer(&buffer);

                buffer.copy_within((i + digit)..(i + 2 * digit), i);
                print_buffer(&buffer);

                last = last
                    .max(i + 2 * digit + (last - (i + 1 + digit)))
                    .max(i + digit + ((i + 1 + digit) - (i + 1)))
                    .max(i + ((i + 2 * digit) - (i + digit)));
            }
            Char::Letter(_) => {
                i += 1;
            }
            Char::Unused => break,
        }
    }
    println!("\nResult:");
    print_buffer(&buffer);
    Ok(())
}

Test 1

Group: 1, 2

Verdict: WRONG ANSWER

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 1, 2

Verdict: WRONG ANSWER

Test 5

Group: 1, 2

Verdict: WRONG ANSWER

Test 6

Group: 1, 2

Verdict: WRONG ANSWER

Test 7

Group: 1, 2

Verdict: WRONG ANSWER

Test 8

Group: 2

Verdict: OUTPUT LIMIT EXCEEDED

Test 9

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'range end index 144951 out of range for slice of length 50000',...

Test 10

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'range end index 145051 out of range for slice of length 50000',...

Test 11

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'range end index 500001 out of range for slice of length 50000',...

Test 12

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'range end index 375001 out of range for slice of length 50000',...

Test 13

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'range end index 328126 out of range for slice of length 50000',...

Test 14

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'range end index 310886 out of range for slice of length 50000',...