Submission #59856274

---

Source Code Expand

use std::collections::HashMap;

use proconio::{input, marker::Chars};

pub fn main() {
    input! {
        n: usize,
        k: usize,
        s: Chars
    };
    let mut connected = -1;
    let mut map = HashMap::new();
    let mut sep = vec![];
    for i in 0..n {
        if s[i] == '1' {
            if connected == -1 {
                connected = i as i64;
                sep.push(i as i64);
            }
        } else {
            if connected != -1 {
                map.insert(connected, i as i64);
            }
            connected = -1;
        }
    }
    if connected != -1 {
        map.insert(connected, n as i64);
    }
    let mut ans = "".to_string();
    let mut count = 0;
    let mut i = 0;
    loop {
        if i == n as i64 {
            break;
        }
        if let Some(&v) = map.get(&(i as i64)) {
            count += 1;
            if count == k {
                for _ in 0..(v - i as i64) {
                    i += 1;
                }
            }
            for _ in 0..(v - i as i64) {
                ans.push('1');
                i += 1;
            }
            if count == k - 1 {
                if let Some(&v) = map.get(&(sep[k - 1])) {
                    for _ in 0..(v - sep[k - 1]) {
                        ans.push('1');
                    }
                }
            }
        } else {
            ans.push('0');
            i += 1;
        }
    }
    println!("{}", ans);
}

Submission Info

Judge Result