結果

問題 No.576 E869120 and Rings
ユーザー snteasntea
提出日時 2018-05-21 11:41:45
言語 Rust
(1.77.0 + proconio)
結果
TLE  
実行時間 -
コード長 3,935 bytes
コンパイル時間 13,217 ms
コンパイル使用メモリ 384,172 KB
実行使用メモリ 22,468 KB
最終ジャッジ日時 2024-06-28 14:55:17
合計ジャッジ時間 23,137 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 TLE -
testcase_02 TLE -
testcase_03 TLE -
testcase_04 TLE -
testcase_05 TLE -
testcase_06 TLE -
testcase_07 TLE -
testcase_08 TLE -
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 TLE -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused macro definition: `printf`
  --> src/main.rs:82:18
   |
82 |     macro_rules! printf { ($($arg:tt)*) => (write!(out, $($arg)*).unwrap()); }
   |                  ^^^^^^
   |
   = note: `#[warn(unused_macros)]` on by default

warning: unused variable: `n`
  --> src/main.rs:85:10
   |
85 |     let (n, k) = readl!(usize, usize);
   |          ^ help: if this is intentional, prefix it with an underscore: `_n`
   |
   = note: `#[warn(unused_variables)]` on by default

ソースコード

diff #

#[allow(dead_code)]
fn getline() -> String {
    let mut res = String::new();
    std::io::stdin().read_line(&mut res).ok();
    res
}

#[allow(unused_macros)]
macro_rules! readl {
    ($t: ty) => {
        {
            let s = getline();
            s.trim().parse::<$t>().unwrap()
        }
    };
    ($( $t: ty),+ ) => {
        {
            let s = getline();
            let mut iter = s.trim().split(' ');
            ($(iter.next().unwrap().parse::<$t>().unwrap(),)*)
        }
    };
}

#[allow(unused_macros)]
macro_rules! readlvec {
    ($t:ty) => {{
        let s = getline();
        let iter = s.trim().split(' ');
        iter.map(|x| x.parse().unwrap()).collect::<Vec<$t>>()
    }};
}

#[allow(unused_macros)]
macro_rules! debug {
    ($x:expr) => {
        println!("{}: {:?}", stringify!($x), $x)
    };
}
// macro_rules! debug { ($x: expr) => () }

#[allow(dead_code)]
fn show<T>(iter: T) -> String
where
    T: Iterator,
    T::Item: std::fmt::Display,
{
    let mut res = iter.fold(String::new(), |sum, e| sum + &format!("{} ", e));
    res.pop();
    res
}

#[allow(unused_imports)]
use std::cmp::{max, min};
#[allow(unused_imports)]
use std::collections::btree_map::Entry::{Occupied, Vacant};
#[allow(unused_imports)]
use std::collections::{BTreeMap, BTreeSet, BinaryHeap, HashMap, HashSet, VecDeque};

#[derive(PartialEq, PartialOrd, Debug)]
struct Ordered<T: std::cmp::PartialOrd>(T);

impl<T> std::cmp::Eq for Ordered<T>
where
    T: std::cmp::PartialOrd,
{
}

impl<T> std::cmp::Ord for Ordered<T>
where
    T: std::cmp::PartialOrd,
{
    fn cmp(&self, other: &Self) -> std::cmp::Ordering {
        self.partial_cmp(other).unwrap()
    }
}

fn main() {
    use std::io::Write;
    let out = std::io::stdout();
    let mut out = std::io::BufWriter::new(out.lock());
    macro_rules! printf { ($($arg:tt)*) => (write!(out, $($arg)*).unwrap()); }
    macro_rules! printfln { () => (writeln!(out).unwrap()); ($($arg:tt)*) => (writeln!(out, $($arg)*).unwrap()); }

    let (n, k) = readl!(usize, usize);
    let a: Vec<i32> = readl!(String)
        .into_bytes()
        .into_iter()
        .map(|x| if x == b'0' { 0 } else { 1 })
        .collect();
    let check = |x| {
        // debug!(x);
        let mut set = VecDeque::new();
        set.push_back(0.);
        for &e in &a {
            let b = *set.back().unwrap();
            set.push_back(b + e as f64 - x);
        }
        // debug!(set);
        let mut os = BTreeMap::new();
        for e in set.iter().skip(k).cloned().map(Ordered) {
            *os.entry(e).or_insert(0) += 1;
        }

        for &e in &a {
            // debug!(set);
            // debug!(os);
            {
                let (&Ordered(max), _) = os.iter().rev().next().unwrap();
                let max = max;

                // debug!((max, *set.front().unwrap()));
                if max > *set.front().unwrap() {
                    return false;
                }
            }
            // os.remove(&Ordered(*set.get(k).unwrap()));
            *os.entry(Ordered(*set.get(k).unwrap())).or_insert(-1) -= 1;
            if *os.get(&Ordered(*set.get(k).unwrap())).unwrap() == 0 {
                os.remove(&Ordered(*set.get(k).unwrap()));
            }
            // debug!(os);
            set.pop_front();
            let b = *set.back().unwrap();
            // os.insert(Ordered(b + e as f64 - x));
            *os.entry(Ordered(b + e as f64 - x)).or_insert(0) += 1;
            // debug!(os);
            set.push_back(b + e as f64 - x);
        }
        true
    };
    // for e in 0..100 {
    //     let mut a = 0.01 * e as f64;
    //     debug!(check(a));
    // }
    // debug!(check(0.9));
    let mut l = 0.;
    let mut r = 1.;
    for _ in 0..10 {
        let m = (l + r) / 2.;
        let res = check(m);
        // debug!(res);
        if res {
            r = m;
        } else {
            l = m;
        }
    }
    printfln!("{}", r);
}
0