#259576 (Rust) No.576 E869120 and Rings

• 問題: No.576 E869120 and Rings
• ユーザー: sntea
• 提出日時: 2018-05-22 03:06:19
• 言語: Rust (1.77.0 + proconio)
• 結果: AC
• 実行時間: 131 ms / 1,500 ms
• コード長: 5,568 bytes
• コンパイル時間: 15,265 ms
• コンパイル使用メモリ: 378,636 KB
• 実行使用メモリ: 7,980 KB
• 最終ジャッジ日時: 2024-06-28 15:42:43
• 合計ジャッジ時間: 16,460 ms
• ジャッジサーバーID （参考情報）: judge5 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	100 ms; 7,848 KB
testcase_01	AC	92 ms; 7,844 KB
testcase_02	AC	111 ms; 7,848 KB
testcase_03	AC	113 ms; 7,720 KB
testcase_04	AC	98 ms; 7,848 KB
testcase_05	AC	98 ms; 7,848 KB
testcase_06	AC	89 ms; 7,844 KB
testcase_07	AC	112 ms; 7,844 KB
testcase_08	AC	102 ms; 7,844 KB
testcase_09	AC	89 ms; 7,848 KB
testcase_10	AC	102 ms; 7,848 KB
testcase_11	AC	86 ms; 7,848 KB
testcase_12	AC	84 ms; 7,980 KB
testcase_13	AC	84 ms; 7,976 KB
testcase_14	AC	131 ms; 7,844 KB
testcase_15	AC	29 ms; 7,848 KB
testcase_16	AC	81 ms; 7,844 KB
testcase_17	AC	107 ms; 7,948 KB
testcase_18	AC	86 ms; 7,852 KB
testcase_19	AC	85 ms; 7,976 KB
testcase_20	AC	69 ms; 7,848 KB
testcase_21	AC	87 ms; 7,848 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	1 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	1 ms; 5,376 KB
testcase_29	AC	1 ms; 5,376 KB

コンパイルメッセージ

warning: unused macro definition: `printf`
   --> src/main.rs:144:18
    |
144 |     macro_rules! printf { ($($arg:tt)*) => (write!(out, $($arg)*).unwrap()); }
    |                  ^^^^^^
    |
    = note: `#[warn(unused_macros)]` on by default

warning: function `sliding_minimum` is never used
  --> src/main.rs:53:4
   |
53 | fn sliding_minimum<T>(a: &Vec<T>, width: usize) -> Vec<T>
   |    ^^^^^^^^^^^^^^^
   |
   = note: `#[warn(dead_code)]` on by default

warning: method `unwrap` is never used
   --> src/main.rs:118:8
    |
117 | impl<T: PartialOrd> Ordered<T> {
    | ------------------------------ method in this implementation
118 |     fn unwrap(self) -> T {
    |        ^^^^^^

ソースコード

#[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
}

fn sliding_minimum<T>(a: &Vec<T>, width: usize) -> Vec<T>
where
    T: std::cmp::PartialOrd + Clone
{
    let mut res = Vec::with_capacity(a.len() - width);
    let mut set = std::collections::VecDeque::new();

    for (i, e) in a.iter().enumerate().take(width) {
        while let Some((v, i)) = set.pop_back() {
            if v < e {
                set.push_back((v, i));
                break;
            }
        }
        set.push_back((e, i));
    }
    for (s, e) in a.iter().skip(width).enumerate() {
        res.push(set.front().expect("sliding_minimum error").0.clone());
        while let Some((v, i)) = set.pop_back() {
            if v < e {
                set.push_back((v, i));
                break;
            }
        }
        set.push_back((e, s + width));
        while let Some((v, i)) = set.pop_front() {
            if s + 1 <= i {
                set.push_front((v, i));
                break;
            }
        }
    }
    res.push(set.front().expect("sliding_minimum error").0.clone());

    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, Clone)]
struct Ordered<T: PartialOrd>(T);

impl<T> Eq for Ordered<T>
where
    T: PartialOrd + Clone,
{
}

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

impl<T: Copy + PartialOrd> Copy for Ordered<T> {}

impl<T: PartialOrd> Ordered<T> {
    fn unwrap(self) -> T {
        let Ordered(res) = self;
        res
    }
}

#[derive(Eq, PartialEq, Clone, Debug)]
struct Reverse<T>(T);

impl<T: PartialOrd> PartialOrd for Reverse<T> {
    fn partial_cmp(&self, other: &Reverse<T>) -> Option<std::cmp::Ordering> {
        other.0.partial_cmp(&self.0)
    }
}

impl<T: Ord> Ord for Reverse<T> {
    fn cmp(&self, other: &Reverse<T>) -> std::cmp::Ordering {
        other.0.cmp(&self.0)
    }
}


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();
    
    
    
    if n == 1 {
        printfln!("{}", a[0]);
        return;
    }
    // let check = |x: f64| {
    //     let mut a: Vec<_> = vec![Reverse(Ordered(0.))].into_iter()
    //         .chain(
    //             a.iter().cloned()
    //             .chain(a.iter().cloned())
    //             .map(|e| Reverse(Ordered(e as f64 - x)))
    //         ).collect();
    //     for i in 0..a.len() - 1 {
    //         (a[i + 1].0).0 += (a[i].0).0;
    //     }
    //     let maxi: Vec<_> = sliding_minimum(
    //         &a,
    //         n - k,
    //     ).into_iter().map(|Reverse(Ordered(x))| x).collect();
    //     for (Reverse(Ordered(l)), &rmax) in a.into_iter().take(n+1).zip(maxi.iter().cycle().skip(k)) { 
    //         if l < rmax {
    //             return false;
    //         }
    //     }
    //     true
    // };
    // let check = |h| {
    //     let mut min = -1e15;
    //     let mut cur = 0;
    //     let cum = vec![0; n*2+1];
    //     for (i, j) in 
    //     true
    // };
    let a: Vec<_> = a.iter().cloned().chain(a.iter().cloned()).collect();
    let check = |x| {
        let mut r = 0.;
        let mut l = 0.;
        let mut minl = 1e99;
        let mut li = 0;
        for ri in 0..n*2 {
            while li + k <= ri {
                if minl > l {
                    minl = l;
                }
                l += a[li] as f64 - x;
                li += 1;
            }
            if r > minl {
                return false;
            }
            r += a[ri] as f64 - x;
        }
        true
    };
    let mut l = 0.;
    let mut r = 1.;
    for _ in 0..30 {
        let m = (l + r) / 2.;
        let res = check(m);
        // debug!(res);
        if res {
            r = m;
        } else {
            l = m;
        }
    }

    printfln!("{}", (l+r)/2.);
}