結果

問題 No.1471 Sort Queries
ユーザー Masaki Kitaguchi
提出日時 2022-02-02 23:25:18
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 16 ms / 2,000 ms
コード長 1,805 bytes
コンパイル時間 11,649 ms
コンパイル使用メモリ 379,688 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-11 09:38:12
合計ジャッジ時間 12,920 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

#[macro_export]
macro_rules! setup {
    { mut $input:ident: SplitWhitespace $(,)? } => {
        use std::io::Read;
        let mut buf = String::new();
        std::io::stdin().read_to_string(&mut buf).ok();
        let mut $input = buf.split_whitespace();
    };
}

#[macro_export]
macro_rules! parse_next {
    ($str_iter:expr) => {
        $str_iter.next().unwrap().parse().ok().unwrap()
    };
}

#[allow(unused_imports)]
use std::collections::{HashMap, HashSet, VecDeque};

fn main() {
    setup! { mut input: SplitWhitespace };

    let _n: usize = parse_next!(input);
    let q: usize = parse_next!(input);
    let s: String = parse_next!(input);
    let mut l: Vec<usize> = vec![];
    let mut r: Vec<usize> = vec![];
    let mut x: Vec<usize> = vec![];
    (0..q).for_each(|_| {
        l.push(parse_next!(input));
        r.push(parse_next!(input));
        x.push(parse_next!(input));
    });

    let mut alphs_acc = vec![];
    {
        alphs_acc.push(vec![0; 26]);
    }
    for (i, c) in s.chars().enumerate() {
        let mut next = alphs_acc[i].clone();
        next[(c as u8 - 'a' as u8) as usize] += 1;
        alphs_acc.push(next);
    }

    for i in 0..q {
        let left = alphs_acc[l[i] - 1].clone();
        let right = alphs_acc[(r[i] - 1) + 1].clone();

        let mut alphs_lr = vec![0; 26];
        for c in 'a'..='z' {
            let idx = (c as u8 - 'a' as u8) as usize;
            alphs_lr[idx] = right[idx] - left[idx];
        }

        let ans = (|| {
            let mut count = 0;
            for c in 'a'..='z' {
                count += alphs_lr[(c as u8 - 'a' as u8) as usize];
                if x[i] <= count {
                    return c;
                }
            }
            unreachable!();
        })();

        println!("{}", ans);
    }
}
0