QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#349231#8335. Fast Hash Transformucup-team296#AC ✓1021ms32676kbRust32.4kb2024-03-10 00:09:412024-03-10 00:09:41

Judging History

你现在查看的是最新测评结果

  • [2024-03-10 00:09:41]
  • 评测
  • 测评结果:AC
  • 用时:1021ms
  • 内存:32676kb
  • [2024-03-10 00:09:41]
  • 提交

answer

// 
pub mod solution {
//{"name":"h","group":"Manual","url":"","interactive":false,"timeLimit":2000,"tests":[{"input":"","output":""}],"testType":"single","input":{"type":"stdin","fileName":null,"pattern":null},"output":{"type":"stdout","fileName":null,"pattern":null},"languages":{"java":{"taskClass":"h"}}}

use std::time::Instant;

#[allow(unused)]
use crate::dbg;
use crate::algo_lib::io::input::Input;
use crate::algo_lib::io::output::Output;
use crate::algo_lib::misc::gen_vector::gen_vec;
use crate::algo_lib::misc::rand::Random;
use crate::algo_lib::seg_trees::lazy_seg_tree::SegTree;
use crate::algo_lib::seg_trees::seg_tree_trait::SegTreeNode;

#[derive(Clone, Copy)]
struct F {
    xor: u64,
    matrix: [u64; 64],
}

impl Default for F {
    fn default() -> Self {
        Self::identity()
    }
}

impl F {
    pub fn identity() -> Self {
        let mut matrix = [0; 64];
        for i in 0..64 {
            matrix[i] = 1 << i;
        }
        Self { xor: 0, matrix }
    }

    pub fn join(&self, other: &Self) -> F {
        let xor = other.apply(self.xor);
        let mut matrix = [0; 64];
        for i in 0..64 {
            let mut tmp = 0;
            for j in 0..64 {
                // if (self.matrix[i] >> j) & 1 == 1 {
                //     matrix[i] ^= other.matrix[j];
                // }
                let coef = (self.matrix[i] >> j) & 1;
                tmp ^= other.matrix[j] * coef;
            }
            matrix[i] = tmp;
        }
        F { xor, matrix }
    }

    pub fn apply(&self, x: u64) -> u64 {
        let mut res = self.xor;
        for i in 0..64 {
            if (x >> i) & 1 == 1 {
                res ^= self.matrix[i];
            }
        }
        res
    }

    pub fn gen_random(rnd: &mut Random) -> Self {
        let mut xor = rnd.gen_u64();
        let mut matrix = [0; 64];
        for i in 0..64 {
            matrix[i] = rnd.gen_u64();
        }
        F { xor, matrix }
    }
}

fn read_f(input: &mut Input) -> F {
    let mut xor = 0;
    let mut matrix = [0; 64];
    let num_turms = input.usize();
    for _i in 0..num_turms {
        let shift = input.usize();
        let op = input.usize();
        let aj = input.u64();
        for bit in 0..64 {
            let nbit = (bit + shift) % 64;
            if op == 0 {
                // or
                if ((aj >> nbit) & 1) == 0 {
                    matrix[bit] ^= 1 << nbit;
                } else {
                    xor ^= 1 << nbit;
                }
            } else {
                assert_eq!(op, 1);
                // and
                if ((aj >> nbit) & 1) == 1 {
                    matrix[bit] ^= 1 << nbit;
                }
            }
        }
    }
    xor ^= input.u64();
    F { xor, matrix }
}

impl SegTreeNode for F {
    fn join_nodes(l: &Self, r: &Self, context: &Self::Context) -> Self {
        l.join(r)
    }

    fn apply_update(node: &mut Self, update: &Self::Update) {
        todo!()
    }

    fn join_updates(current: &mut Self::Update, add: &Self::Update) {
        todo!()
    }

    type Update = ();

    type Context = ();
}

fn solve(input: &mut Input, out: &mut Output, _test_case: usize) {
    let n = input.usize();
    let q = input.usize();
    let _c = input.usize();
    let g = gen_vec(n, |_| read_f(input));
    let mut st = SegTree::new(n, |i| g[i]);
    for _ in 0..q {
        let op = input.usize();
        if op == 0 {
            let l = input.usize() - 1;
            let r = input.usize();
            // let base = st.get(l..r);
            // let mut base = F::identity();
            // for i in l..r {
            //     base = base.join(&g[i]);
            // }
            let mut x = input.u64();
            st.visit(l..r, &mut |f| {
                x = f.apply(x);
            });
            // let res = base.apply(x);
            out.println(x);
        } else {
            assert_eq!(op, 1);
            let pos = input.usize() - 1;
            st.update_point(pos, read_f(input));
            // g[pos] = read_f(input);
        }
    }
}

fn stress() {
    for it in 1.. {
        let mut rnd = Random::new(it);
        dbg!(it);
        let n = 20_000;

        let start = Instant::now();

        let g = gen_vec(n, |_| F::gen_random(&mut rnd));
        let mut st = SegTree::new(n, |i| g[i]);
        for _ in 0..n {
            let l = rnd.gen(0..n);
            let r = rnd.gen(l + 1..n + 1);
            // let base = st.get(l..r);
            let mut x = rnd.gen_u64();
            st.visit(l..r, &mut |f| {
                x = f.apply(x);
            });
        }
        dbg!(start.elapsed());
    }
}

pub(crate) fn run(mut input: Input, mut output: Output) -> bool {
    solve(&mut input, &mut output, 1);
    output.flush();
    true
}

}
pub mod algo_lib {
pub mod io {
pub mod input {
use std::fmt::Debug;
use std::io::Read;
use std::marker::PhantomData;
use std::path::Path;
use std::str::FromStr;

pub struct Input {
    input: Box<dyn Read>,
    buf: Vec<u8>,
    at: usize,
    buf_read: usize,
}

macro_rules! read_integer_fun {
    ($t:ident) => {
        #[allow(unused)]
        pub fn $t(&mut self) -> $t {
            self.read_integer()
        }
    };
}

impl Input {
    const DEFAULT_BUF_SIZE: usize = 4096;

    ///
    /// Using with stdin:
    /// ```no_run
    /// use algo_lib::io::input::Input;
    /// let stdin = std::io::stdin();
    /// let input = Input::new(Box::new(stdin));
    /// ```
    ///
    /// For read files use ``new_file`` instead.
    ///
    ///
    pub fn new(input: Box<dyn Read>) -> Self {
        Self {
            input,
            buf: vec![0; Self::DEFAULT_BUF_SIZE],
            at: 0,
            buf_read: 0,
        }
    }

    pub fn new_stdin() -> Self {
        let stdin = std::io::stdin();
        Self::new(Box::new(stdin))
    }

    pub fn new_file<P: AsRef<Path>>(path: P) -> Self {
        let file = std::fs::File::open(&path)
            .unwrap_or_else(|_| panic!("Can't open file: {:?}", path.as_ref().as_os_str()));
        Self::new(Box::new(file))
    }

    pub fn new_with_size(input: Box<dyn Read>, buf_size: usize) -> Self {
        Self {
            input,
            buf: vec![0; buf_size],
            at: 0,
            buf_read: 0,
        }
    }

    pub fn new_file_with_size<P: AsRef<Path>>(path: P, buf_size: usize) -> Self {
        let file = std::fs::File::open(&path)
            .unwrap_or_else(|_| panic!("Can't open file: {:?}", path.as_ref().as_os_str()));
        Self::new_with_size(Box::new(file), buf_size)
    }

    pub fn get(&mut self) -> Option<u8> {
        if self.refill_buffer() {
            let res = self.buf[self.at];
            self.at += 1;
            Some(res)
        } else {
            None
        }
    }

    pub fn peek(&mut self) -> Option<u8> {
        if self.refill_buffer() {
            Some(self.buf[self.at])
        } else {
            None
        }
    }

    pub fn skip_whitespace(&mut self) {
        while let Some(b) = self.peek() {
            if !char::from(b).is_whitespace() {
                return;
            }
            self.get();
        }
    }

    pub fn next_token(&mut self) -> Option<Vec<u8>> {
        self.skip_whitespace();
        let mut res = Vec::new();
        while let Some(c) = self.get() {
            if char::from(c).is_whitespace() {
                break;
            }
            res.push(c);
        }
        if res.is_empty() {
            None
        } else {
            Some(res)
        }
    }

    //noinspection RsSelfConvention
    pub fn is_exhausted(&mut self) -> bool {
        self.peek().is_none()
    }

    pub fn has_more_elements(&mut self) -> bool {
        !self.is_exhausted()
    }

    pub fn read<T: Readable>(&mut self) -> T {
        T::read(self)
    }

    pub fn vec<T: Readable>(&mut self, size: usize) -> Vec<T> {
        let mut res = Vec::with_capacity(size);
        for _ in 0usize..size {
            res.push(self.read());
        }
        res
    }

    pub fn string_vec(&mut self, size: usize) -> Vec<Vec<u8>> {
        let mut res = Vec::with_capacity(size);
        for _ in 0usize..size {
            res.push(self.string());
        }
        res
    }

    pub fn read_line(&mut self) -> String {
        let mut res = String::new();
        while let Some(c) = self.get() {
            if c == b'\n' {
                break;
            }
            if c == b'\r' {
                if self.peek() == Some(b'\n') {
                    self.get();
                }
                break;
            }
            res.push(c.into());
        }
        res
    }

    #[allow(clippy::should_implement_trait)]
    pub fn into_iter<T: Readable>(self) -> InputIterator<T> {
        InputIterator {
            input: self,
            phantom: Default::default(),
        }
    }

    fn read_integer<T: FromStr>(&mut self) -> T
    where
        <T as FromStr>::Err: Debug,
    {
        let res = self.read_string();
        res.parse::<T>().unwrap()
    }

    fn read_string(&mut self) -> String {
        match self.next_token() {
            None => {
                panic!("Input exhausted");
            }
            Some(res) => unsafe { String::from_utf8_unchecked(res) },
        }
    }

    pub fn string_as_string(&mut self) -> String {
        self.read_string()
    }

    pub fn string(&mut self) -> Vec<u8> {
        self.read_string().into_bytes()
    }

    fn read_char(&mut self) -> char {
        self.skip_whitespace();
        self.get().unwrap().into()
    }

    fn read_float(&mut self) -> f64 {
        self.read_string().parse().unwrap()
    }

    pub fn f64(&mut self) -> f64 {
        self.read_float()
    }

    fn refill_buffer(&mut self) -> bool {
        if self.at == self.buf_read {
            self.at = 0;
            self.buf_read = self.input.read(&mut self.buf).unwrap();
            self.buf_read != 0
        } else {
            true
        }
    }

    read_integer_fun!(i32);
    read_integer_fun!(i64);
    read_integer_fun!(i128);
    read_integer_fun!(u32);
    read_integer_fun!(u64);
    read_integer_fun!(usize);
}

pub trait Readable {
    fn read(input: &mut Input) -> Self;
}

impl Readable for String {
    fn read(input: &mut Input) -> Self {
        input.read_string()
    }
}

impl Readable for char {
    fn read(input: &mut Input) -> Self {
        input.read_char()
    }
}

impl Readable for f64 {
    fn read(input: &mut Input) -> Self {
        input.read_string().parse().unwrap()
    }
}

impl Readable for f32 {
    fn read(input: &mut Input) -> Self {
        input.read_string().parse().unwrap()
    }
}

impl<T: Readable> Readable for Vec<T> {
    fn read(input: &mut Input) -> Self {
        let size = input.read();
        input.vec(size)
    }
}

pub struct InputIterator<T: Readable> {
    input: Input,
    phantom: PhantomData<T>,
}

impl<T: Readable> Iterator for InputIterator<T> {
    type Item = T;

    fn next(&mut self) -> Option<Self::Item> {
        self.input.skip_whitespace();
        self.input.peek().map(|_| self.input.read())
    }
}

macro_rules! read_integer {
    ($t:ident) => {
        impl Readable for $t {
            fn read(input: &mut Input) -> Self {
                input.read_integer()
            }
        }
    };
}

read_integer!(i8);
read_integer!(i16);
read_integer!(i32);
read_integer!(i64);
read_integer!(i128);
read_integer!(isize);
read_integer!(u8);
read_integer!(u16);
read_integer!(u32);
read_integer!(u64);
read_integer!(u128);
read_integer!(usize);
}
pub mod output {
use std::io::Write;

pub struct Output {
    output: Box<dyn Write>,
    buf: Vec<u8>,
    at: usize,
    auto_flush: bool,
}

impl Output {
    const DEFAULT_BUF_SIZE: usize = 4096;

    pub fn new(output: Box<dyn Write>) -> Self {
        Self {
            output,
            buf: vec![0; Self::DEFAULT_BUF_SIZE],
            at: 0,
            auto_flush: false,
        }
    }

    pub fn new_stdout() -> Self {
        let stdout = std::io::stdout();
        Self::new(Box::new(stdout))
    }

    pub fn new_file(path: impl AsRef<std::path::Path>) -> Self {
        let file = std::fs::File::create(path).unwrap();
        Self::new(Box::new(file))
    }

    pub fn new_with_auto_flush(output: Box<dyn Write>) -> Self {
        Self {
            output,
            buf: vec![0; Self::DEFAULT_BUF_SIZE],
            at: 0,
            auto_flush: true,
        }
    }

    pub fn flush(&mut self) {
        if self.at != 0 {
            self.output.write_all(&self.buf[..self.at]).unwrap();
            self.at = 0;
            self.output.flush().expect("Couldn't flush output");
        }
    }

    pub fn print<T: Writable>(&mut self, s: T) {
        s.write(self);
    }

    pub fn println<T: Writable>(&mut self, s: T) {
        s.write(self);
        self.put(b'\n');
    }

    pub fn put(&mut self, b: u8) {
        self.buf[self.at] = b;
        self.at += 1;
        if self.at == self.buf.len() {
            self.flush();
        }
    }

    pub fn maybe_flush(&mut self) {
        if self.auto_flush {
            self.flush();
        }
    }

    pub fn print_per_line<T: Writable>(&mut self, arg: &[T]) {
        for i in arg {
            i.write(self);
            self.put(b'\n');
        }
    }

    pub fn print_iter<T: Writable, I: Iterator<Item = T>>(&mut self, iter: I) {
        let mut first = true;
        for e in iter {
            if first {
                first = false;
            } else {
                self.put(b' ');
            }
            e.write(self);
        }
    }

    pub fn print_iter_ref<'a, T: 'a + Writable, I: Iterator<Item = &'a T>>(&mut self, iter: I) {
        let mut first = true;
        for e in iter {
            if first {
                first = false;
            } else {
                self.put(b' ');
            }
            e.write(self);
        }
    }
}

impl Write for Output {
    fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
        let mut start = 0usize;
        let mut rem = buf.len();
        while rem > 0 {
            let len = (self.buf.len() - self.at).min(rem);
            self.buf[self.at..self.at + len].copy_from_slice(&buf[start..start + len]);
            self.at += len;
            if self.at == self.buf.len() {
                self.flush();
            }
            start += len;
            rem -= len;
        }
        if self.auto_flush {
            self.flush();
        }
        Ok(buf.len())
    }

    fn flush(&mut self) -> std::io::Result<()> {
        self.flush();
        Ok(())
    }
}

pub trait Writable {
    fn write(&self, output: &mut Output);
}

impl Writable for &str {
    fn write(&self, output: &mut Output) {
        output.write_all(self.as_bytes()).unwrap();
    }
}

impl Writable for String {
    fn write(&self, output: &mut Output) {
        output.write_all(self.as_bytes()).unwrap();
    }
}

impl Writable for char {
    fn write(&self, output: &mut Output) {
        output.put(*self as u8);
    }
}

impl<T: Writable> Writable for [T] {
    fn write(&self, output: &mut Output) {
        output.print_iter_ref(self.iter());
    }
}

impl<T: Writable> Writable for Vec<T> {
    fn write(&self, output: &mut Output) {
        self[..].write(output);
    }
}

macro_rules! write_to_string {
    ($t:ident) => {
        impl Writable for $t {
            fn write(&self, output: &mut Output) {
                self.to_string().write(output);
            }
        }
    };
}

write_to_string!(u8);
write_to_string!(u16);
write_to_string!(u32);
write_to_string!(u64);
write_to_string!(u128);
write_to_string!(usize);
write_to_string!(i8);
write_to_string!(i16);
write_to_string!(i32);
write_to_string!(i64);
write_to_string!(i128);
write_to_string!(isize);
write_to_string!(f32);
write_to_string!(f64);

impl<T: Writable, U: Writable> Writable for (T, U) {
    fn write(&self, output: &mut Output) {
        self.0.write(output);
        output.put(b' ');
        self.1.write(output);
    }
}

impl<T: Writable, U: Writable, V: Writable> Writable for (T, U, V) {
    fn write(&self, output: &mut Output) {
        self.0.write(output);
        output.put(b' ');
        self.1.write(output);
        output.put(b' ');
        self.2.write(output);
    }
}
}
}
pub mod misc {
pub mod dbg_macro {
#[macro_export]
#[allow(unused_macros)]
macro_rules! dbg {
    ($first_val:expr, $($val:expr),+ $(,)?) => {
        eprint!("[{}:{}] {} = {:?}",
                    file!(), line!(), stringify!($first_val), &$first_val);
        ($(eprint!(", {} = {:?}", stringify!($val), &$val)),+,);
        eprintln!();
    };
    ($first_val:expr) => {
        eprintln!("[{}:{}] {} = {:?}",
                    file!(), line!(), stringify!($first_val), &$first_val)
    };
}
}
pub mod gen_vector {
pub fn gen_vec<T>(n: usize, f: impl FnMut(usize) -> T) -> Vec<T> {
    (0..n).map(f).collect()
}
}
pub mod num_traits {
use std::cmp::Ordering;
use std::fmt::Debug;
use std::ops::Add;
use std::ops::AddAssign;
use std::ops::Div;
use std::ops::DivAssign;
use std::ops::Mul;
use std::ops::MulAssign;
use std::ops::Sub;
use std::ops::SubAssign;

pub trait HasConstants<T> {
    const MAX: T;
    const MIN: T;
    const ZERO: T;
    const ONE: T;
    const TWO: T;
}

pub trait ConvSimple<T> {
    fn from_i32(val: i32) -> T;
    fn to_i32(self) -> i32;
    fn to_f64(self) -> f64;
}

pub trait Signum {
    fn signum(&self) -> i32;
}

pub trait Number:
    Copy
    + Add<Output = Self>
    + AddAssign
    + Sub<Output = Self>
    + SubAssign
    + Mul<Output = Self>
    + MulAssign
    + Div<Output = Self>
    + DivAssign
    + PartialOrd
    + PartialEq
    + HasConstants<Self>
    + Default
    + Debug
    + Sized
    + ConvSimple<Self>
{
}

impl<
        T: Copy
            + Add<Output = Self>
            + AddAssign
            + Sub<Output = Self>
            + SubAssign
            + Mul<Output = Self>
            + MulAssign
            + Div<Output = Self>
            + DivAssign
            + PartialOrd
            + PartialEq
            + HasConstants<Self>
            + Default
            + Debug
            + Sized
            + ConvSimple<Self>,
    > Number for T
{
}

macro_rules! has_constants_impl {
    ($t: ident) => {
        impl HasConstants<$t> for $t {
            // TODO: remove `std` for new rust version..
            const MAX: $t = std::$t::MAX;
            const MIN: $t = std::$t::MIN;
            const ZERO: $t = 0;
            const ONE: $t = 1;
            const TWO: $t = 2;
        }

        impl ConvSimple<$t> for $t {
            fn from_i32(val: i32) -> $t {
                val as $t
            }

            fn to_i32(self) -> i32 {
                self as i32
            }

            fn to_f64(self) -> f64 {
                self as f64
            }
        }
    };
}

has_constants_impl!(i32);
has_constants_impl!(i64);
has_constants_impl!(i128);
has_constants_impl!(u32);
has_constants_impl!(u64);
has_constants_impl!(u128);
has_constants_impl!(usize);
has_constants_impl!(u8);

impl ConvSimple<Self> for f64 {
    fn from_i32(val: i32) -> Self {
        val as f64
    }

    fn to_i32(self) -> i32 {
        self as i32
    }

    fn to_f64(self) -> f64 {
        self
    }
}

impl HasConstants<Self> for f64 {
    const MAX: Self = Self::MAX;
    const MIN: Self = -Self::MAX;
    const ZERO: Self = 0.0;
    const ONE: Self = 1.0;
    const TWO: Self = 2.0;
}

impl<T: Number + Ord> Signum for T {
    fn signum(&self) -> i32 {
        match self.cmp(&T::ZERO) {
            Ordering::Greater => 1,
            Ordering::Less => -1,
            Ordering::Equal => 0,
        }
    }
}
}
pub mod rand {
use crate::algo_lib::misc::gen_vector::gen_vec;
use crate::algo_lib::misc::num_traits::Number;
use std::ops::Range;
use std::time::SystemTime;
use std::time::UNIX_EPOCH;

#[allow(dead_code)]
pub struct Random {
    state: u64,
}

impl Random {
    pub fn gen_u64(&mut self) -> u64 {
        let mut x = self.state;
        x ^= x << 13;
        x ^= x >> 7;
        x ^= x << 17;
        self.state = x;
        x
    }

    #[allow(dead_code)]
    pub fn next_in_range(&mut self, from: usize, to: usize) -> usize {
        assert!(from < to);
        (from as u64 + self.gen_u64() % ((to - from) as u64)) as usize
    }

    pub fn gen_index<T>(&mut self, a: &[T]) -> usize {
        self.gen(0..a.len())
    }

    #[allow(dead_code)]
    #[inline(always)]
    pub fn gen_double(&mut self) -> f64 {
        (self.gen_u64() as f64) / (std::usize::MAX as f64)
    }

    #[allow(dead_code)]
    pub fn new(seed: u64) -> Self {
        let state = if seed == 0 { 787788 } else { seed };
        Self { state }
    }

    pub fn new_time_seed() -> Self {
        let time = SystemTime::now();
        let seed = (time.duration_since(UNIX_EPOCH).unwrap().as_nanos() % 1_000_000_000) as u64;
        if seed == 0 {
            Self::new(787788)
        } else {
            Self::new(seed)
        }
    }

    #[allow(dead_code)]
    pub fn gen_permutation(&mut self, n: usize) -> Vec<usize> {
        let mut result: Vec<_> = (0..n).collect();
        for i in 0..n {
            let idx = self.next_in_range(0, i + 1);
            result.swap(i, idx);
        }
        result
    }

    pub fn shuffle<T>(&mut self, a: &mut [T]) {
        for i in 1..a.len() {
            a.swap(i, self.gen(0..i + 1));
        }
    }

    pub fn gen<T>(&mut self, range: Range<T>) -> T
    where
        T: Number,
    {
        let from = T::to_i32(range.start);
        let to = T::to_i32(range.end);
        assert!(from < to);
        let len = (to - from) as usize;
        T::from_i32(self.next_in_range(0, len) as i32 + from)
    }

    pub fn gen_vec<T>(&mut self, n: usize, range: Range<T>) -> Vec<T>
    where
        T: Number,
    {
        gen_vec(n, |_| self.gen(range.clone()))
    }

    pub fn gen_nonempty_range(&mut self, n: usize) -> Range<usize> {
        let x = self.gen(0..n);
        let y = self.gen(0..n);
        if x <= y {
            x..y + 1
        } else {
            y..x + 1
        }
    }

    pub fn gen_bool(&mut self) -> bool {
        self.gen(0..2) == 0
    }
}
}
}
pub mod seg_trees {
pub mod lazy_seg_tree {
use std::ops::Range;

use crate::algo_lib::seg_trees::seg_tree_trait::SegTreeNode;

///
/// Segment Tree
///
#[allow(unused)]
#[derive(Clone)]
pub struct SegTree<T: SegTreeNode> {
    n: usize,
    tree: Vec<T>,
    updates_to_push: Vec<Option<T::Update>>,
    context: T::Context,
    right_nodes: Vec<usize>,
}

#[allow(unused)]
impl<T: SegTreeNode> SegTree<T> {
    fn pull(&mut self, v: usize, vr: usize) {
        self.tree[v] = T::join_nodes(&self.tree[v + 1], &self.tree[vr], &self.context);
    }

    fn build(&mut self, v: usize, l: usize, r: usize, init_val: &T) {
        if l + 1 == r {
            self.tree[v] = init_val.clone();
        } else {
            let m = (l + r) >> 1;
            let vr = v + ((m - l) << 1);
            self.build(v + 1, l, m, init_val);
            self.build(vr, m, r, init_val);
            self.pull(v, vr);
        }
    }

    fn push(&mut self, v: usize, l: usize, r: usize) {
        let update = self.updates_to_push[v].clone();
        self.updates_to_push[v] = None;
        match update {
            None => {}
            Some(update) => {
                let m = (l + r) >> 1;
                self.apply_update(v + 1, &update, m - l == 1);
                self.apply_update(v + ((r - l) & !1), &update, r - m == 1);
            }
        }
    }

    fn get_(&mut self, v: usize, l: usize, r: usize, ql: usize, qr: usize) -> T {
        assert!(qr >= l);
        assert!(ql < r);
        if ql <= l && r <= qr {
            return self.tree[v].clone();
        }
        let m = (l + r) >> 1;
        let vr = v + ((m - l) << 1);
        self.push(v, l, r);
        let res = if ql >= m {
            self.get_(vr, m, r, ql, qr)
        } else if qr <= m {
            self.get_(v + 1, l, m, ql, qr)
        } else {
            T::join_nodes(
                &self.get_(v + 1, l, m, ql, qr),
                &self.get_(vr, m, r, ql, qr),
                &self.context,
            )
        };
        self.pull(v, vr);
        res
    }

    fn visit_(
        &mut self,
        v: usize,
        l: usize,
        r: usize,
        ql: usize,
        qr: usize,
        f: &mut impl FnMut(&T),
    ) {
        assert!(qr >= l);
        assert!(ql < r);
        if ql <= l && r <= qr {
            f(&self.tree[v]);
            return;
        }
        let m = (l + r) >> 1;
        let vr = v + ((m - l) << 1);
        // self.push(v, l, r);
        if ql >= m {
            self.visit_(vr, m, r, ql, qr, f);
        } else if qr <= m {
            self.visit_(v + 1, l, m, ql, qr, f)
        } else {
            self.visit_(v + 1, l, m, ql, qr, f);
            self.visit_(vr, m, r, ql, qr, f);
        };
        // self.pull(v, vr);
        // res
    }

    fn join_updates(current: &mut Option<T::Update>, add: &T::Update) {
        match current {
            None => *current = Some(add.clone()),
            Some(current) => T::join_updates(current, add),
        };
    }

    fn apply_update(&mut self, v: usize, update: &T::Update, is_leaf: bool) {
        T::apply_update(&mut self.tree[v], update);
        if !is_leaf {
            Self::join_updates(&mut self.updates_to_push[v], update);
        }
    }

    fn modify_(&mut self, v: usize, l: usize, r: usize, ql: usize, qr: usize, update: &T::Update) {
        assert!(qr >= l);
        assert!(ql < r);
        if ql <= l && r <= qr {
            self.apply_update(v, update, r - l == 1);
            return;
        }
        let m = (l + r) >> 1;
        let vr = v + ((m - l) << 1);
        self.push(v, l, r);
        if ql >= m {
            self.modify_(vr, m, r, ql, qr, update);
        } else if qr <= m {
            self.modify_(v + 1, l, m, ql, qr, update);
        } else {
            self.modify_(v + 1, l, m, ql, qr, update);
            self.modify_(vr, m, r, ql, qr, update);
        };
        self.pull(v, vr);
    }

    pub fn update(&mut self, range: Range<usize>, update: T::Update) {
        if range.is_empty() {
            return;
        }
        assert!(!range.is_empty());
        self.modify_(0, 0, self.n, range.start, range.end, &update);
    }

    pub fn update_point(&mut self, pos: usize, new_node: T) {
        let mut l = 0;
        let mut r = self.n;
        let mut v: usize = 0;
        let mut to_pull = vec![];
        while r - l > 1 {
            let m = (l + r) >> 1;
            let vr = v + ((m - l) << 1);
            self.push(v, l, r);
            to_pull.push((v, vr));
            if pos < m {
                r = m;
                v = v + 1;
            } else {
                l = m;
                v = vr;
            }
        }
        self.tree[v] = new_node;
        for (v, vr) in to_pull.into_iter().rev() {
            self.pull(v, vr);
        }
    }

    fn find_last_true_(
        &mut self,
        v: usize,
        l: usize,
        r: usize,
        range: Range<usize>,
        f: &impl Fn(&T) -> bool,
    ) -> Option<usize> {
        if range.start >= r || l >= range.end {
            return None;
        }
        let m = (l + r) >> 1;
        let vr = v + ((m - l) << 1);
        if range.start <= l && r <= range.end {
            if !f(&self.tree[v]) {
                return None;
            }
            if r - l == 1 {
                return Some(l);
            }
        }
        self.push(v, l, r);
        if let Some(res) = self.find_last_true_(vr, m, r, range.clone(), f) {
            Some(res)
        } else {
            self.find_last_true_(v + 1, l, m, range, f)
        }
    }

    // returns position
    pub fn find_last_true(&mut self, range: Range<usize>, f: impl Fn(&T) -> bool) -> Option<usize> {
        self.find_last_true_(0, 0, self.n, range, &f)
    }

    pub fn get(&mut self, range: Range<usize>) -> T {
        if range.is_empty() {
            return T::default();
        }
        self.get_(0, 0, self.n, range.start, range.end)
    }

    pub fn visit(&mut self, range: Range<usize>, f: &mut impl FnMut(&T)) {
        if range.is_empty() {
            return;
        }
        self.visit_(0, 0, self.n, range.start, range.end, f);
    }

    pub fn new_with_context(n: usize, f: impl Fn(usize) -> T, context: T::Context) -> Self {
        assert!(n > 0);
        let tree = vec![T::default(); 2 * n - 1];
        let updates_to_push = vec![None; 2 * n - 1];
        let mut res = SegTree {
            n,
            tree,
            updates_to_push,
            context,
            right_nodes: vec![],
        };
        res.build_f(0, 0, n, &f);
        res
    }

    pub fn new(n: usize, f: impl Fn(usize) -> T) -> Self
    where
        T::Context: Default,
    {
        assert!(n > 0);
        let tree = vec![T::default(); 2 * n - 1];
        let updates_to_push = vec![None; 2 * n - 1];
        let mut res = SegTree {
            n,
            tree,
            updates_to_push,
            context: T::Context::default(),
            right_nodes: vec![],
        };
        res.build_f(0, 0, n, &f);
        res
    }

    fn build_f(&mut self, v: usize, l: usize, r: usize, f: &impl Fn(usize) -> T) {
        if l + 1 == r {
            self.tree[v] = f(l);
        } else {
            let m = (l + r) >> 1;
            let vr = v + ((m - l) << 1);
            self.build_f(v + 1, l, m, f);
            self.build_f(vr, m, r, f);
            self.pull(v, vr);
        }
    }

    pub fn len(&self) -> usize {
        self.n
    }

    pub fn expert_get_node(&self, node: usize) -> &T {
        &self.tree[node]
    }

    pub fn expert_get_left_node(&self, node: usize) -> usize {
        node + 1
    }

    fn build_right_nodes(&mut self, v: usize, l: usize, r: usize) {
        if l + 1 == r {
            self.right_nodes.push(0);
        } else {
            let m = (l + r) >> 1;
            let vr = v + ((m - l) << 1);
            self.right_nodes.push(vr);
            self.build_right_nodes(v + 1, l, m);
            self.build_right_nodes(vr, m, r);
        }
    }

    // TODO: shouldn't be mut
    pub fn expert_get_right_node(&mut self, node: usize) -> usize {
        if self.right_nodes.is_empty() {
            self.build_right_nodes(0, 0, self.n);
        }
        self.right_nodes[node]
    }

    // Used for Kinetic Seg Tree
    pub fn expert_rebuild_nodes(&mut self, should_rebuild: impl Fn(&T, &T::Context) -> bool) {
        self.expert_rebuild_nodes_(0, 0, self.n, &should_rebuild);
    }

    fn expert_rebuild_nodes_(
        &mut self,
        v: usize,
        l: usize,
        r: usize,
        should_rebuild: &impl Fn(&T, &T::Context) -> bool,
    ) {
        if r - l <= 1 || !should_rebuild(&self.tree[v], &self.context) {
            return;
        }
        let m = (l + r) >> 1;
        let vr = v + ((m - l) << 1);
        self.push(v, l, r);

        self.expert_rebuild_nodes_(v + 1, l, m, should_rebuild);
        self.expert_rebuild_nodes_(vr, m, r, should_rebuild);

        self.pull(v, vr);
    }

    pub fn update_context(&mut self, f: impl Fn(&mut T::Context)) {
        f(&mut self.context);
    }

    pub fn get_context(&self) -> &T::Context {
        &self.context
    }
}
}
pub mod seg_tree_trait {
pub trait SegTreeNode: Clone + Default {
    fn join_nodes(l: &Self, r: &Self, context: &Self::Context) -> Self;

    fn apply_update(node: &mut Self, update: &Self::Update);
    fn join_updates(current: &mut Self::Update, add: &Self::Update);

    type Update: Clone;
    type Context;
}
}
}
}
fn main() {
    let input = algo_lib::io::input::Input::new_stdin();
    let mut output = algo_lib::io::output::Output::new_stdout();
    crate::solution::run(input, output);
}

这程序好像有点Bug,我给组数据试试?

详细

Test #1:

score: 100
Accepted
time: 0ms
memory: 2136kb

input:

3 5 1
1 4 0 0 51966
1 60 0 0 0
1 0 0 16 15
0 1 1 771
0 2 2 32368
0 3 3 0
1 2 2 0 0 15 61 1 4095 46681
0 1 3 2023

output:

64206
2023
31
1112

result:

ok 4 tokens

Test #2:

score: 0
Accepted
time: 1ms
memory: 2228kb

input:

9 9 3
32 9 0 17785061119123981789 33 0 10890571864137198682 42 0 9437574736788763477 34 0 5239651887868507470 55 0 14741743279679654187 27 1 1444116632918569317 38 1 5740886562180922636 1 1 8113356142324084796 3 0 10955266306442425904 60 0 16421026339459788005 53 0 1595107134632608917 48 1 923204972...

output:

9487331362121050549
3906661590723083106
15757672015979182109
4975471776251039345
11503109206538591140
3763610618439604410

result:

ok 6 tokens

Test #3:

score: 0
Accepted
time: 13ms
memory: 2192kb

input:

1 20000 400
32 13 0 1721926083061553294 52 1 8951352260297008058 6 0 3180917732545757991 63 1 14978562500072226750 50 1 7331113732303115313 59 0 688182721924779475 12 0 16291922173168822489 61 0 16018198440613086698 8 0 12494084957448674305 7 0 2834422858291562646 42 1 10354539547309738529 28 0 2541...

output:

11827781865759498816
7454610526276185721
9581050724293785387
2177163806257271094
14004004964877510141
18073834598135159471
16966489063087641088
12289032565388413023
17823140805867698239
18104549908461644670
15570008264282957124
12400954982104000299
9842549278742638708
16535034933613060362
1561642006...

result:

ok 19600 tokens

Test #4:

score: 0
Accepted
time: 114ms
memory: 2676kb

input:

500 20000 400
32 3 0 9869926173615303101 39 1 11114680792832491178 54 1 3380955246053990760 31 0 16868042247314276464 26 0 5814925615581342395 30 1 1114053898154397400 46 1 9215698002668459992 38 1 12938485987410997250 58 0 8030873196223549640 0 0 16055471402053138912 47 1 16568729207788187629 63 0 ...

output:

9119093329811149961
16901643057538871933
17161855998497876349
3964234071281411558
13588188063229334268
15557968976322375381
4612345875926431452
9507168112801039022
9504318642653891468
217407202160767706
12982350345598971306
17957502630817476223
6353877977318728572
15552768639781831485
16778108770682...

result:

ok 19600 tokens

Test #5:

score: 0
Accepted
time: 241ms
memory: 8304kb

input:

4000 20000 400
35 33 0 18435679328748604368 55 1 10851974578636476759 1 0 11332084644969697080 13 0 4243547822701774011 19 0 18197854269436975495 32 0 10133703694198056054 6 0 12655387670867301210 36 0 1246525872821095171 51 1 812047498663608637 4 0 9797423115860097390 7 1 12105773148377740641 17 0 ...

output:

11875257514484243925
3443357416933857062
16160011677622853538
1582145987019406393
15019762274690743371
3128972641411454448
10632018957963074870
2420532366876270818
16130728863118353230
15834956073901517645
18404809296474853851
10982435108266120760
16463778300806795274
11990886156320593058
1145171640...

result:

ok 19600 tokens

Test #6:

score: 0
Accepted
time: 610ms
memory: 32468kb

input:

20000 20000 0
34 47 1 3147866938814566873 50 0 8051884074279018250 4 0 11476150812073861567 54 0 3931985566612211642 60 1 9226417006726638292 49 0 2435425653073267226 33 1 5976119177961927073 40 1 3169532703977184656 2 1 17206894689684881943 37 0 2316971949450684490 7 1 7087775905790436416 18 1 7557...

output:

8031710763259664674
10015579400510819759
9509776159199873854
252965904282343862
17471441301398284397
6167329408972068582
11581702001320217920
13373488743211628824
2094753313448112669
15503010008451014749
384500896248723935
10501371892025480221
8907735695899875922
14479597201387282763
164403466075406...

result:

ok 20000 tokens

Test #7:

score: 0
Accepted
time: 619ms
memory: 32452kb

input:

20000 20000 20
28 31 1 17220760822712602145 12 1 10079395927654210001 40 0 10440736241216457314 20 1 14759495678166748212 55 1 8734257463550073646 60 0 543206106562221008 29 1 5402811237936853387 52 1 3884345269948184760 22 0 7873959847686200341 15 1 18396630536251250330 25 0 18230407003294263406 14...

output:

6531775129959975384
6212576544894999781
4191848452578359691
2769536540387251859
15526337103142577854
14948743844803225542
15235110724610778185
9004056994453026335
1028305510694260706
13496210650896843548
13961471020487846633
1864980030930734934
15243868808579626755
10451839696548403150
1178402342726...

result:

ok 19980 tokens

Test #8:

score: 0
Accepted
time: 703ms
memory: 32420kb

input:

20000 20000 400
41 15 1 10590708978689078436 33 0 17448869030270552656 37 1 16782453056389226553 2 1 18313039076194285622 53 1 7894371271572806769 60 1 14563226108042670650 56 0 12694119759311053234 12 1 969626878679760122 28 1 8906626075909573228 20 1 11632670066953088447 50 0 13097960756795495550 ...

output:

7425391644666486729
17533666397961516801
16986235811843827275
1784742314571007240
13192305384063626572
12739810377012216000
1179361465141596122
7698346401428161235
6903188112913915716
5380404381348976227
16126105607866972637
12798978320947566556
11234201442491665890
16073897288956866956
151328474491...

result:

ok 19600 tokens

Test #9:

score: 0
Accepted
time: 701ms
memory: 32676kb

input:

20000 20000 400
33 39 1 17067623245236507261 27 1 7041428814521205530 50 1 10823426118594256003 28 1 7163716190894912799 12 1 4080987667516350158 63 0 17082717673883070565 17 0 11310350135715835231 51 1 12855244004029414317 38 0 9814237273168847221 57 1 3708701962235763971 37 0 10158992933772396697 ...

output:

6864973236048047224
18318008901523164537
13500746067907696382
13161681605750995854
3452654261090196316
14847903013724109682
7301818645657195470
15784097910646013208
6555334273152043996
6337001136120562705
7065460407919669838
17502323856909932125
12099828260978288865
17244785354672463736
159661862214...

result:

ok 19600 tokens

Test #10:

score: 0
Accepted
time: 555ms
memory: 32456kb

input:

20000 20000 0
37 46 0 4806156443854081866 29 0 6910842714881233745 61 0 9379366064412681113 32 1 718568893402460472 45 0 1234243654449881049 16 0 9791590151480029686 24 1 801156398497308107 20 1 1638149966892153162 3 1 483739892768149714 56 1 3070030763953269690 38 1 11944075913457601606 6 1 8068547...

output:

17693343388614420171
11014279187501816246
7111154205373939902
5948421254644613369
5776121468606637836
16944170640450069348
8394185836099893155
11947149219582604015
4508739183749291929
11471060687727420580
3924131475517252887
1743542114579130111
14487529569441993654
8062193838630657668
18359613799309...

result:

ok 20000 tokens

Test #11:

score: 0
Accepted
time: 649ms
memory: 32432kb

input:

20000 20000 400
31 63 1 14360706182574306953 17 0 4643864577315796645 48 0 11264878137122897405 18 1 14150130986659920202 25 1 15979000250901513596 49 0 16241841209566112679 37 1 16762565151400121253 14 1 7376170230332808198 26 1 10868082441744868454 27 1 6949308347230687639 44 1 4116452321258930556...

output:

4493451016206578040
14208826853413050113
15020158700931574670
16337826900074673926
5403566933376608394
8871156492968482557
8911109963819675601
6213157285507240354
17190717170193641517
15578273901773478953
1369444627312020075
11786462107951385516
17634527799358234224
18347358352139830828
145863906383...

result:

ok 19600 tokens

Test #12:

score: 0
Accepted
time: 635ms
memory: 32456kb

input:

20000 20000 0
25 16 0 2668205375195949736 34 0 2748287585311204102 37 1 4531486636255948251 24 0 14410309392802368907 52 1 851885811168352867 47 1 15887239727531457455 42 0 8819527325570258215 44 0 16146066124743535517 46 1 1041563265194349313 11 1 13140073107604291185 0 1 16670532562293262804 56 1 ...

output:

5924012028700061898
4718073419648080016
13993322115865028469
82790239609178342
887419913876033685
15321668567642867070
8962935781265467660
1552533755174937777
16683793257623944188
6900756788022393102
10981237528745871227
5789630421744738481
9056874037200094100
15328577526113324947
627381852022728881...

result:

ok 20000 tokens

Test #13:

score: 0
Accepted
time: 738ms
memory: 32460kb

input:

20000 20000 400
29 26 0 4544814232153615705 62 0 13471881549916225637 53 0 595040484360290534 17 1 11949377632182068615 20 0 9311349995021422035 60 0 816595034603135343 48 0 10654197142859256352 24 0 4772788762907504538 54 0 15584542718632709463 19 1 2151387765439259665 41 1 15099291996319444694 40 ...

output:

1423325416659742802
17371677980372930727
3681232738426467215
13266462173687709877
12639633063779823269
1946619485256865431
12989302207328517036
14138867084917472527
18218337902938347758
3372796243270362661
673871038008779791
9077527952253879051
7326631285358366273
2710349874806590369
172928358344422...

result:

ok 19600 tokens

Test #14:

score: 0
Accepted
time: 716ms
memory: 32508kb

input:

20000 20000 400
24 38 1 3460586314685566112 26 0 4188084273268315686 61 0 1227557406397452582 25 1 5747999803901747386 41 1 1907261769878407698 27 0 16752772615002344376 34 0 17112817990633067537 60 0 9291256903378353577 45 0 7510343555807629464 13 0 18007693761515887577 9 1 17317953029923040761 4 0...

output:

13100999329968562920
15516017614708332089
5382331705592343945
357522576585990254
16311520569626827168
6758986479859611544
12732461424037837989
8966988217248420501
10391550114259677068
4660694210255306341
7237506373169380284
13657244350225493605
6916780657676036471
10881113620462446827
16277857127600...

result:

ok 19600 tokens

Test #15:

score: 0
Accepted
time: 984ms
memory: 32512kb

input:

20000 20000 400
60 25 1 2719522369398288789 40 1 9400902170286318935 6 1 9521944178235051324 43 0 11768204916287391421 22 0 12726538799306592512 47 1 15759776307217345226 17 1 15438840904724459733 13 0 17863856648581711698 29 1 4032777103500438360 10 0 683992519125165540 26 1 15461667428831774672 14...

output:

412280358023687627
14769812881817125733
18318455003071307239
6658808483284331159
6130439376668456888
1492308137069243960
5853920885317257980
12553163529022332915
7520793755811132601
13993258994649409340
13568418050081351467
12309096149487021368
17899306611786296579
2598853739059100346
14630776750608...

result:

ok 19600 tokens

Test #16:

score: 0
Accepted
time: 979ms
memory: 32480kb

input:

20000 20000 400
58 17 0 7751036624596546188 29 0 17432650278695392920 63 0 4389503693497138241 24 0 11063030485327807810 45 1 18240337466752243882 59 1 17804018980843534887 60 1 5872699360277748939 21 1 10429471649278268372 27 0 16762274761588446397 54 0 6030940442592696599 19 1 13270942932118095691...

output:

17867517807501868664
7775943633465469655
1824462793515136478
12630456144448858727
16944355600951673184
14837233611662521712
13878709289450326681
13750017221938869139
11379793111096427897
15527971528797740116
5872004578520784281
11280146030218952435
5218412287620909707
15541801824852151484
5650476389...

result:

ok 19600 tokens

Test #17:

score: 0
Accepted
time: 1000ms
memory: 32552kb

input:

20000 20000 400
60 22 0 12707603241789460389 46 0 15086313545825117890 33 0 909121147706895901 59 1 6114670732197551336 41 0 3090389396605293219 63 0 6608018621123394175 38 1 11836608073091750746 14 1 4878457553941336535 5 0 6024711164477768229 25 1 67199414342206654 24 0 1139176812912408779 16 0 12...

output:

11101670153251608253
675780991220106254
15866623643054791619
16323951405331282505
9135544362908319645
7642151295897109981
12351493946367393308
1935066719605622920
7518368202469961257
11600515247827283279
15933103715396571729
13453007077105135208
14727385649041622999
123215011875209372
76221879547507...

result:

ok 19600 tokens

Test #18:

score: 0
Accepted
time: 1021ms
memory: 32564kb

input:

20000 20000 400
57 0 1 16995927798044259033 30 1 4411529108320693455 35 1 14740826024996968953 32 1 14741500464787789772 5 1 13621297821910096766 47 1 1805557230674866983 26 0 1852515218899978614 37 1 14167448543730803554 15 0 8207408801869448845 7 0 2253659179521891807 61 1 1303777112793499927 1 1 ...

output:

14640221015002441272
8108797686286238378
619977752116985977
17860455208859938460
2219391733727955287
17098130710123326231
4643402762732727695
1576822124091279449
2112594951252396904
11012866398108256228
3100264803360198048
1520325785935749501
17234063909328734373
2294517371241459639
6577965043160831...

result:

ok 19600 tokens

Test #19:

score: 0
Accepted
time: 1018ms
memory: 32388kb

input:

20000 20000 400
60 23 1 12967402093469929995 29 0 12482782180176810891 47 1 17290909632526370536 5 0 17372530581982291607 62 1 13987764289696466564 41 1 8421162609706963610 53 0 18089202028338115523 10 0 1312033850971950221 2 0 3337291528457528731 18 1 17751876270582349954 32 0 13359684730271699557 ...

output:

8219634040467306280
7488593258162917054
3429645423088159837
13823280993584110417
4972072459402131521
8504404100763034378
5815021261728531941
5670841473953742448
6721982245071008063
8353993923949852878
4277531481899017404
5173775653727609205
4061296038432070306
11044359884601198871
272114767395246212...

result:

ok 19600 tokens

Test #20:

score: 0
Accepted
time: 1016ms
memory: 32484kb

input:

20000 20000 400
56 3 0 14386867255986651070 17 1 5876678743420202175 24 0 2472800002233203764 40 0 3974575279492546522 5 1 10323896862538344788 31 0 15302550669828857297 10 0 8188514003112427229 28 0 9350793473465284653 34 1 1051624389221640716 56 1 12992224832956122800 0 1 12521917684359350214 33 1...

output:

16023098103166573969
5488057777512126106
2364974952009032490
6867156023979603961
13898097818261510588
3896697852262723116
13075557814956081246
550932186457628322
1501306951093128873
12801415593941350129
12431218450278358855
8287212554931924015
7521994383511852994
7781645210267880075
1658905057583045...

result:

ok 19600 tokens

Extra Test:

score: 0
Extra Test Passed