結果

問題 No.2546 Many Arithmetic Sequences
ユーザー akakimidoriakakimidori
提出日時 2023-11-25 07:27:42
言語 Rust
(1.77.0)
結果
WA  
実行時間 -
コード長 5,502 bytes
コンパイル時間 1,304 ms
コンパイル使用メモリ 193,880 KB
実行使用メモリ 15,272 KB
最終ジャッジ日時 2023-11-25 07:27:49
合計ジャッジ時間 5,860 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,676 KB
testcase_01 AC 1 ms
6,676 KB
testcase_02 AC 1 ms
6,676 KB
testcase_03 AC 53 ms
6,676 KB
testcase_04 AC 96 ms
13,268 KB
testcase_05 AC 81 ms
8,960 KB
testcase_06 AC 17 ms
6,676 KB
testcase_07 AC 38 ms
6,676 KB
testcase_08 AC 12 ms
6,676 KB
testcase_09 AC 31 ms
8,192 KB
testcase_10 AC 58 ms
12,540 KB
testcase_11 AC 62 ms
14,348 KB
testcase_12 AC 29 ms
6,676 KB
testcase_13 AC 58 ms
8,704 KB
testcase_14 AC 53 ms
13,516 KB
testcase_15 AC 15 ms
6,676 KB
testcase_16 AC 51 ms
11,664 KB
testcase_17 AC 83 ms
8,960 KB
testcase_18 AC 43 ms
8,064 KB
testcase_19 AC 51 ms
7,296 KB
testcase_20 AC 28 ms
7,168 KB
testcase_21 AC 40 ms
6,676 KB
testcase_22 AC 63 ms
8,704 KB
testcase_23 AC 123 ms
15,032 KB
testcase_24 AC 126 ms
15,028 KB
testcase_25 AC 123 ms
15,036 KB
testcase_26 AC 132 ms
15,032 KB
testcase_27 AC 120 ms
15,036 KB
testcase_28 AC 102 ms
15,268 KB
testcase_29 AC 102 ms
15,268 KB
testcase_30 AC 102 ms
15,268 KB
testcase_31 AC 103 ms
15,268 KB
testcase_32 AC 102 ms
15,272 KB
testcase_33 AC 1 ms
6,676 KB
testcase_34 AC 73 ms
15,216 KB
testcase_35 AC 1 ms
6,676 KB
testcase_36 AC 86 ms
13,868 KB
testcase_37 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn main() {
    input! {
        n: usize,
        m: usize,
        p: [(i64, i64); n],
    }
    let mut h = std::collections::BinaryHeap::new();
    let mut line = vec![];
    for (a, d) in p {
        if d < 0 {
            h.push((a, d));
        } else {
            // k * a + k * (k - 1) / 2 * d
            // k 確定 2a + (k - 1)d
            line.push((-d, -(2 * a - d)));
        }
    }
    if h.is_empty() {
        let m = m as i64;
        let ans = line.iter().map(|(a, d)| a * m + m * (m - 1) / 2 * d).max().unwrap();
        println!("{}", ans);
        return;
    }
    let mut neg = vec![0; m + 1];
    for i in 1..(m + 1) {
        let (a, d) = h.pop().unwrap();
        neg[i] = neg[i - 1] + a;
        h.push((a + d, d));
    }
    if line.is_empty() {
        println!("{}", neg[m]);
        return;
    }
    let cht = ConvexHullTrick::new(line);
    let mut ans = std::i64::MIN;
    for (k, neg) in neg.iter().enumerate() {
        let k = (m - k) as i64;
        let v = cht.find(k);
        ans = ans.max(*neg + k * -v / 2);
    }
    println!("{}", ans);
}

// ---------- begin input macro ----------
// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
#[macro_export]
macro_rules! input {
    (source = $s:expr, $($r:tt)*) => {
        let mut iter = $s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
    ($($r:tt)*) => {
        let s = {
            use std::io::Read;
            let mut s = String::new();
            std::io::stdin().read_to_string(&mut s).unwrap();
            s
        };
        let mut iter = s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
}

#[macro_export]
macro_rules! input_inner {
    ($iter:expr) => {};
    ($iter:expr, ) => {};
    ($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($iter, $t);
        input_inner!{$iter $($r)*}
    };
}

#[macro_export]
macro_rules! read_value {
    ($iter:expr, ( $($t:tt),* )) => {
        ( $(read_value!($iter, $t)),* )
    };
    ($iter:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
    };
    ($iter:expr, chars) => {
        read_value!($iter, String).chars().collect::<Vec<char>>()
    };
    ($iter:expr, bytes) => {
        read_value!($iter, String).bytes().collect::<Vec<u8>>()
    };
    ($iter:expr, usize1) => {
        read_value!($iter, usize) - 1
    };
    ($iter:expr, $t:ty) => {
        $iter.next().unwrap().parse::<$t>().expect("Parse error")
    };
}
// ---------- end input macro ----------
// 以下のクエリを処理する
//   add_line(a, b): 直線 ax + b を追加
//   find(x): min (ax + b) を返す。空のとき呼ぶとREになる
// 計算量
// 直線追加クエリがN回飛んでくるとする
// 直線追加 償却O(log N)
// 点質問: O((log N)^2)
// ---------- begin incremental convex hull trick (min) ----------
// reference: https://yukicoder.me/wiki/decomposable_searching_problem
// verify: https://old.yosupo.jp/submission/35150
#[derive(Clone)]
struct ConvexHullTrick {
    line: Vec<(i64, i64)>,
}

impl ConvexHullTrick {
    fn new(mut line: Vec<(i64, i64)>) -> Self {
        assert!(line.len() > 0);
        line.sort();
        line.dedup_by(|a, b| a.0 == b.0);
        let mut stack: Vec<(i64, i64)> = vec![];
        for (a, b) in line {
            while stack.len() >= 2 {
                let len = stack.len();
                let (c, d) = stack[len - 1];
                let (e, f) = stack[len - 2];
                let x = (d - b).div_euclid(a - c);
                let y = (f - d).div_euclid(c - e);
                if x >= y {
                    stack.pop();
                } else {
                    break;
                }
            }
            stack.push((a, b));
        }
        ConvexHullTrick { line: stack }
    }
    fn find(&self, x: i64) -> i64 {
        let mut l = 0;
        let mut r = self.line.len() - 1;
        let line = &self.line;
        let func = |k: usize| -> i64 {
            let (a, b) = line[k];
            a * x + b
        };
        while r - l >= 3 {
            let ll = (2 * l + r) / 3;
            let rr = (l + 2 * r) / 3;
            if func(ll) <= func(rr) {
                r = rr;
            } else {
                l = ll;
            }
        }
        line[l..=r].iter().map(|p| p.0 * x + p.1).min().unwrap()
    }
}

#[derive(Clone, Default)]
pub struct IncrementalCHT {
    size: usize,
    cht: Vec<(ConvexHullTrick, usize)>,
}

impl IncrementalCHT {
    pub fn new() -> Self {
        IncrementalCHT {
            size: 0,
            cht: vec![],
        }
    }
    pub fn add_line(&mut self, a: i64, b: i64) {
        self.size += 1;
        let mut line = vec![(a, b)];
        let mut p = 0;
        while self.cht.last().map_or(false, |q| q.1 == p) {
            p += 1;
            line.append(&mut self.cht.pop().unwrap().0.line);
        }
        let cht = ConvexHullTrick::new(line);
        self.cht.push((cht, p));
    }
    pub fn find(&self, x: i64) -> i64 {
        self.cht.iter().map(|p| p.0.find(x)).min().unwrap()
    }
    pub fn append(&mut self, other: &mut Self) {
        if self.size < other.size {
            std::mem::swap(self, other);
        }
        for (mut cht, _) in other.cht.drain(..) {
            for (a, b) in cht.line.drain(..) {
                self.add_line(a, b);
            }
        }
        other.size = 0;
    }
}
// ---------- end incremental convex hull trick (min) ----------
0