結果
問題 |
No.3091 The Little Match Boy
|
ユーザー |
|
提出日時 | 2025-04-07 08:47:44 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 602 ms / 2,000 ms |
コード長 | 5,809 bytes |
コンパイル時間 | 14,375 ms |
コンパイル使用メモリ | 399,104 KB |
実行使用メモリ | 17,856 KB |
最終ジャッジ日時 | 2025-04-07 08:48:16 |
合計ジャッジ時間 | 26,148 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 62 |
ソースコード
#[allow(unused_imports)] use std::{ cell::RefCell, convert::{Infallible, TryFrom, TryInto as _}, fmt::{self, Debug, Display, Formatter,}, fs::{File}, hash::{Hash, Hasher}, iter::{Product, Sum}, marker::PhantomData, ops::{Add, AddAssign, Sub, SubAssign, Div, DivAssign, Mul, MulAssign, Neg, RangeBounds}, str::{FromStr, SplitWhitespace}, sync::{atomic::{self, AtomicU32, AtomicU64}, Once}, collections::{*, btree_map::Range}, mem::{swap}, cmp::{self, Reverse, Ordering, Eq, PartialEq, PartialOrd}, thread::LocalKey, f64::consts::PI, time::Instant, rc::Rc, io::{self, stdin, Read, read_to_string, BufWriter, BufReader, stdout, Write}, }; #[allow(unused_imports)] use proconio::{input, input_interactive, marker::{*}}; #[allow(unused_imports)] //use rand::{thread_rng, Rng, seq::SliceRandom}; #[allow(unused_imports)] //use ac_library::{*}; #[allow(dead_code)] const INF: i64 = 1<<60; #[allow(dead_code)] const MOD: i64 = 998244353; #[allow(dead_code)] const D: [(usize, usize); 4] = [(1, 0), (0, 1), (!0, 0), (0, !0)]; pub trait SegTreeMonoid{ type S: Clone; fn identity()->Self::S; fn op(a: &Self::S, b: &Self::S)->Self::S; } pub struct SegTree<M: SegTreeMonoid>{ n: usize, data: Vec<M::S>, } impl<M: SegTreeMonoid> SegTree<M> { pub fn new(n: usize)->Self{ let n = n.next_power_of_two(); let data = vec![M::identity(); 2*n]; SegTree{ n, data } } pub fn set(&mut self, i: usize, x: M::S){ let mut p = i+self.n; self.data[p] = x; while p > 0{ p /= 2; self.data[p] = M::op(&self.data[p<<1], &self.data[(p<<1)|1]); } } pub fn push(&mut self, i: usize, x: M::S){ let mut p = i+self.n; self.data[p] = M::op(&self.data[p], &x); while p > 0{ p /= 2; self.data[p] = M::op(&self.data[p<<1], &self.data[(p<<1)|1]); } } pub fn prod(&mut self, l: usize, r: usize)->M::S{ let mut p_l = l+self.n; let mut p_r = r+self.n; let mut res_l = M::identity(); let mut res_r = M::identity(); while p_l < p_r{ if p_l&1==1{ res_l = M::op(&res_l, &self.data[p_l]); p_l += 1; } if p_r&1==1{ p_r -= 1; res_r = M::op(&self.data[p_r], &res_r); } p_l >>= 1; p_r >>= 1; } M::op(&res_l, &res_r) } pub fn all_prod(&mut self)->M::S{ self.data[1].clone() } pub fn max_right<F>(&self, mut l: usize, f: F)->usize where F: Fn(&M::S)->bool{ assert!(f(&M::identity())); // これはバグってくれないと多分デバックが悲惨 if l==self.n{return self.n} l += self.n; let mut ac = M::identity(); while { while l%2==0{ l >>= 1; } if !f(&M::op(&ac, &self.data[l])){ while l < self.n{ l <<= 1; let res = M::op(&ac, &self.data[l]); if f(&res){ ac = res; l += 1; } } return l-self.n; } ac = M::op(&ac, &self.data[l]); l += 1; let z = l as isize; (z & -z) != z } {} self.n } pub fn min_left<F>(&self, mut r: usize, f: F)->usize where F: Fn(&M::S)->bool{ assert!(f(&M::identity()));r += self.n;let mut ac = M::identity(); if r==0{return 0} while { while r > 1 && r%2 == 1{ r >>= 1; } if !f(&M::op(&ac, &self.data[r])){ while r < self.n{ r = 2*r+1; let res = M::op(&ac, &self.data[r]); if f(&res){ ac = res; r -= 1; } } return r+1-self.n; } ac = M::op(&ac, &self.data[r]); let z = r as isize; z & -z == z } {} 0 } } pub const B: u128 = 1394107; pub const MO: u128 = (1<<64)-59; pub struct RollingHashMonoid; impl SegTreeMonoid for RollingHashMonoid{ type S = (u128, u128); fn identity() -> Self::S { (1, 0) } fn op(a: &Self::S, b: &Self::S) -> Self::S { let &(b1, h1) = a; let &(b2, h2) = b; ((b1*b2)%MO, (b2*h1+h2)%MO) } } pub fn compress<T: Copy+Ord+std::hash::Hash>(a: &Vec<T>)->Vec<usize>{ let mut b = a.clone(); b.sort(); b.dedup(); let mut dic = std::collections::HashMap::new(); for (i, &v) in b.iter().enumerate(){ dic.insert(v, i); } a.iter().map(|x| dic[x]).collect() } //use proconio::fastout; //#[fastout] fn main(){ input!{ n: usize, m: usize, s: [Usize1; m], } let mut set = HashSet::new(); let mut base = (0..n).collect::<Vec<_>>(); let mut ss = vec![(0, 0); m]; for (i, &idx) in s.iter().enumerate(){ ss[i] = (base[idx], base[idx+1]); base.swap(idx, idx+1); } let mut inv = vec![0; n]; for (i, &v) in base.iter().enumerate(){ inv[v] = i; } let mut segtree = SegTree::<RollingHashMonoid>::new(n); for i in 0..n{ segtree.set(i, (B, base[i] as u128)); } for &(u, v) in &ss{ let (idx1, idx2) = (inv[u], inv[v]); segtree.set(idx2, (B, base[idx1] as u128)); segtree.set(idx1, (B, base[idx2] as u128)); set.insert(segtree.all_prod().1); segtree.set(idx1, (B, base[idx1] as u128)); segtree.set(idx2, (B, base[idx2] as u128)); } println!("{}", set.len()); }