Submission #27662545
Source Code Expand
// Macro by MasuqaT (occar421)
// https://github.com/occar421/ProgrammingContest/tree/master/templates/src/standard_io.rs
#[allow(unused_imports)]
pub use multiset::*;
#[allow(unused_imports)]
pub use point::*;
use std::cmp::*;
use std::collections::*;
use std::fmt::{Debug, Display};
use std::hash::Hash;
#[allow(unused_imports)]
use std::iter::FromIterator;
use std::iter::{Product, Sum};
use std::ops::*;
use std::str::FromStr;
// From https://github.com/tanakh/competitive-rs/blob/d5f51f01a6f85ddbebec4cfcb601746bee727181/src/lib.rs#L1-L92
// and modified by this file author
#[doc(hidden)]
struct Handler<F: FnMut() -> String> {
handle: F,
}
#[doc(hidden)]
macro_rules! prepare_input {
(source = $s:expr) => {{
let mut iter = $s.split_whitespace();
Handler {
handle: || iter.next().unwrap(),
}
}};
(stdin = $s:expr) => {{
let mut bytes = std::io::Read::bytes(std::io::BufReader::new($s));
Handler {
handle: move || {
bytes
.by_ref()
.map(|r| r.unwrap() as char)
.skip_while(|c| c.is_whitespace())
.take_while(|c| !c.is_whitespace())
.collect::<String>()
},
}
}};
}
macro_rules! input_original {
(source = $s:expr; $($r:tt)*) => {
let mut _handler = prepare_input!{source = $s};
let mut _next = || (_handler.handle)();
input_inner!{_next, $($r)*}
};
(stdin = $s:expr; $($r:tt)*) => {
let mut _handler = prepare_input!{stdin = $s};
let mut _next = || (_handler.handle)();
input_inner!{_next, $($r)*}
};
(handler = $h: ident; $($r:tt)*) => {
let mut _next = || ($h.handle)();
input_inner!{_next, $($r)*}
};
}
#[doc(hidden)]
macro_rules! input_inner {
($next:expr) => {};
($next:expr, ) => {};
($next:expr, $var:ident : $t:tt $($r:tt)*) => {
let $var = read_value!($next, $t);
input_inner!{$next $($r)*}
};
($next:expr, mut $var:ident : $t:tt $($r:tt)*) => {
let mut $var = read_value!($next, $t);
input_inner!{$next $($r)*}
};
}
#[doc(hidden)]
macro_rules! read_value {
($next:expr, ( $($t:tt),* )) => {
( $(read_value!($next, $t)),* )
};
($next:expr, [ $t:tt ; $len1:expr ; $len2:expr ]) => {
(0..$len1).map(|_| (0..$len2).map(|_| read_value!($next, $t)).collect::<Vec<_>>()).collect::<Vec<_>>()
};
($next:expr, [ $t:tt ; $len:expr ]) => {
(0..$len).map(|_| read_value!($next, $t)).collect::<Vec<_>>()
};
($next:expr, chars) => {
(read_value!($next, String).chars().collect::<Vec<char>>()) as Vec<char>
};
($next:expr, bytes) => {
(read_value!($next, String).into_bytes()) as Vec<u8>
};
($next:expr, usize0) => {
(read_value!($next, usize)) as usize
};
($next:expr, usize1) => {
(read_value!($next, usize) - 1) as usize
};
($next:expr, (Point2d<$t:tt>)) => {
{
let x = read_value!($next, $t);
let y = read_value!($next, $t);
Point2d { x, y }
}
};
($next:expr, (Point2d<$t:tt>-rev)) => {
{
let y = read_value!($next, $t);
let x = read_value!($next, $t);
Point2d { x, y }
}
};
($next:expr, $t:ty) => {
($next().parse::<$t>().expect("Parse error")) as $t
};
}
#[allow(unused_macros)]
macro_rules! assert_judge {
($method:ident, $input:expr, $expected:expr) => {{
let output = assert_judge_with_output!($method, $input);
let expected = $expected
.trim()
.lines()
.map(|l| l.trim())
.collect::<Vec<_>>()
.join("\n");
assert_eq!(output, expected);
}};
}
#[allow(unused_macros)]
macro_rules! assert_judge_with_output {
($method:ident, $input:expr) => {{
let input = $input.as_bytes();
let mut output = Vec::new();
$method(&input[..], &mut output).expect("Should not emit error");
String::from_utf8(output)
.expect("Not UTF-8")
.trim()
.lines()
.map(|l| l.trim())
.collect::<Vec<_>>()
.join("\n")
}};
}
#[allow(unused_macros)]
#[macro_export]
macro_rules! assert_eq_with_error {
($left:expr, $right:expr, $precision:expr) => {{
match (&$left, &$right, &$precision) {
(left_val, right_val, precision_val) => {
if !(*left_val - *precision_val < *right_val
&& *right_val < *left_val + *precision_val)
{
// The re-borrows below are intentional. Without them, the stack slot for the
// borrow is initialized even before the values are compared, leading to a
// noticeable slow down.
panic!(
r#"assertion failed: `(left == right) +- precision`
left: `{:?}`,
right: `{:?}`,
precision: `{:?}`"#,
&*left_val, &*right_val, &*precision_val
)
}
}
}
}};
}
#[allow(unused_macros)]
macro_rules! assert_judge_with_error {
($method:ident, $input:expr, $expected:expr, $t:ty | $precision:expr ) => {{
let output = assert_judge_with_output!($method, $input);
let actual: $t = output.parse().unwrap();
let expected: $t = $expected.parse().unwrap();
assert_eq_with_error!(actual, expected, $precision);
}};
}
macro_rules! impl_collection_util {
($tr: tt :: $met: ident -> $ret: tt where $req: tt { $proc: ident }, []) => {};
($tr: tt :: $met: ident -> $ret: tt where $req: tt { $proc: ident }, [ $t: tt $(, $r:tt)* ]) => {
impl<T, PT> $tr for $t<PT>
where
T: $req,
PT: $tr<Result = T>,
{
type Result = T;
#[inline]
fn $met(&self) -> $ret<Self::Result> {
$proc(self)
}
}
impl<T, PT> $tr for &$t<PT>
where
T: $req,
PT: $tr<Result = T>,
{
type Result = T;
#[inline]
fn $met(&self) -> $ret<Self::Result> {
$proc(*self)
}
}
impl_collection_util!($tr::$met -> $ret where $req {$proc}, [ $( $r ),* ]);
};
}
type Slice<T> = [T];
type Id<T> = T;
pub trait Min: PartialMin {
fn min(&self) -> Self::Result;
}
pub trait PartialMin {
type Result;
fn partial_min(&self) -> Option<Self::Result>;
}
fn iter_partial_min<'a, T, PT, I>(iter: I) -> Option<T>
where
T: Ord,
PT: 'a + PartialMin<Result = T>,
I: 'a + IntoIterator<Item = &'a PT>,
{
iter.into_iter().filter_map(|x| x.partial_min()).min()
}
impl_collection_util!(
PartialMin::partial_min -> Option where Ord { iter_partial_min },
[Option, Slice, Vec, HashSet]
);
pub fn min_with_partial<T>(o1: Option<T>, o2: Option<T>) -> Option<T>
where
T: Ord,
{
match (o1, o2) {
(Some(v1), Some(v2)) => min(v1, v2).into(),
(o1, None) => o1,
(None, o2) => o2,
}
}
#[allow(unused_macros)]
#[macro_export]
macro_rules! min {
($x: expr) => (Min::min(&$x));
($x: expr, $($z: expr),+) => (::std::cmp::min(Min::min(&$x), min!($($z),*)));
}
#[allow(unused_macros)]
#[macro_export]
macro_rules! partial_min {
($x: expr) => (PartialMin::partial_min(&$x));
($x: expr, $($z: expr),+) => (min_with_partial(PartialMin::partial_min(&$x), partial_min!($($z),*)));
}
pub trait Max: PartialMax {
fn max(&self) -> Self::Result;
}
pub trait PartialMax {
type Result;
fn partial_max(&self) -> Option<Self::Result>;
}
fn iter_partial_max<'a, T, PT, I>(iter: I) -> Option<T>
where
T: Ord,
PT: 'a + PartialMax<Result = T>,
I: 'a + IntoIterator<Item = &'a PT>,
{
iter.into_iter().filter_map(|x| x.partial_max()).max()
}
impl_collection_util!(
PartialMax::partial_max -> Option where Ord { iter_partial_max },
[Option, Slice, Vec, HashSet]
);
pub fn max_with_partial<T>(o1: Option<T>, o2: Option<T>) -> Option<T>
where
T: Ord,
{
match (o1, o2) {
(Some(v1), Some(v2)) => max(v1, v2).into(),
(o1, None) => o1,
(None, o2) => o2,
}
}
#[allow(unused_macros)]
#[macro_export]
macro_rules! max {
($x: expr) => (Max::max(&$x));
($x: expr, $($z: expr),+) => (::std::cmp::max(Max::max(&$x), max!($($z),*)));
}
#[allow(unused_macros)]
#[macro_export]
macro_rules! partial_max {
($x: expr) => (PartialMax::partial_max(&$x));
($x: expr, $($z: expr),+) => (max_with_partial(PartialMax::partial_max(&$x), partial_max!($($z),*)));
}
pub trait AutoSum {
type Result;
fn sum(&self) -> Self::Result;
}
fn iter_auto_sum<'a, T, ST, I>(iter: I) -> T
where
T: Sum,
ST: 'a + AutoSum<Result = T>,
I: 'a + IntoIterator<Item = &'a ST>,
{
iter.into_iter().map(|x| x.sum()).sum()
}
impl_collection_util!(
AutoSum::sum -> Id where Sum { iter_auto_sum },
[Option, Slice, Vec, HashSet]
);
#[allow(unused_macros)]
#[macro_export]
macro_rules! sum {
($x: expr) => (AutoSum::sum(&$x));
($x: expr, $($z: expr),+) => (AutoSum::sum(&$x) + sum!($($z),*));
}
pub trait AutoProduct {
type Result;
fn product(&self) -> Self::Result;
}
fn iter_auto_product<'a, T, ST, I>(iter: I) -> T
where
T: Product,
ST: 'a + AutoProduct<Result = T>,
I: 'a + IntoIterator<Item = &'a ST>,
{
iter.into_iter().map(|x| x.product()).product()
}
impl_collection_util!(
AutoProduct::product -> Id where Product { iter_auto_product },
[Option, Slice, Vec, HashSet]
);
#[allow(unused_macros)]
#[macro_export]
macro_rules! product {
($x: expr) => (AutoProduct::product(&$x));
($x: expr, $($z: expr),+) => (AutoProduct::product(&$x) * product!($($z),*));
}
pub trait GenericInteger:
Copy
+ Clone
+ Eq
+ PartialEq
+ Ord
+ PartialOrd
+ Hash
+ FromStr
+ Display
+ Debug
+ Add<Output = Self>
+ AddAssign
+ Sub<Output = Self>
+ SubAssign
+ Mul<Output = Self>
+ MulAssign
+ Div<Output = Self>
+ DivAssign
+ Rem<Output = Self>
+ RemAssign
{
fn zero() -> Self;
fn one() -> Self;
fn is_odd(&self) -> bool;
fn is_even(&self) -> bool;
}
#[doc(hidden)]
macro_rules! implement_generic_integer {
() => {};
($t:ty $(, $r:ty)*) => {
impl GenericInteger for $t {
#[inline]
fn zero() -> Self { 0 }
#[inline]
fn one() -> Self { 1 }
#[inline]
fn is_odd(&self) -> bool { self % 2 == 1 }
#[inline]
fn is_even(&self) -> bool { self % 2 == 0 }
}
impl PartialMin for $t {
type Result = $t;
#[inline]
fn partial_min(&self) -> Option<Self::Result> {
self.clone().into()
}
}
impl Min for $t {
#[inline]
fn min(&self) -> Self::Result {
self.clone()
}
}
impl PartialMax for $t {
type Result = $t;
#[inline]
fn partial_max(&self) -> Option<Self::Result> {
self.clone().into()
}
}
impl Max for $t {
#[inline]
fn max(&self) -> Self::Result {
self.clone()
}
}
impl AutoSum for $t {
type Result = $t;
#[inline]
fn sum(&self) -> Self {
self.clone()
}
}
impl AutoProduct for $t {
type Result = $t;
#[inline]
fn product(&self) -> Self {
self.clone()
}
}
implement_generic_integer![ $( $r ),* ];
};
}
implement_generic_integer![u8, u16, u32, u64, u128, usize, i8, i16, i32, i64, i128, isize];
#[allow(dead_code)]
pub fn gcd<T>(a: T, b: T) -> T
where
T: GenericInteger,
{
if b == T::zero() {
a
} else {
gcd(b, a % b.clone())
}
}
#[allow(dead_code)]
#[inline]
pub fn lcm<T>(a: T, b: T) -> T
where
T: GenericInteger,
{
gcd_lcm(a, b).1
}
#[inline]
pub fn gcd_lcm<T>(a: T, b: T) -> (T, T)
where
T: GenericInteger,
{
if a == T::zero() && b == T::zero() {
return (T::zero(), T::zero());
}
let gcd = gcd(a, b);
let lcm = a * (b / gcd);
return (gcd, lcm);
}
/// O(√N)
#[allow(dead_code)]
pub fn prime_factorize(n: usize) -> HashMap<usize, usize> {
let mut map = HashMap::new();
let sqrt_n = (n as f64).sqrt().ceil() as usize;
let mut n = n;
for p in 2..=sqrt_n {
if n % p != 0 {
continue;
}
let mut exp_number = 0;
while n % p == 0 {
exp_number += 1;
n /= p;
}
map.insert(p, exp_number);
}
if n != 1 {
map.insert(n, 1);
}
map
}
const INC: [usize; 8] = [4, 2, 4, 2, 4, 6, 2, 6];
// https://memo.sugyan.com/entry/2021/02/06/021949
/// O(N log(logN) )
#[allow(dead_code)]
pub fn eratosthenes_sieve(n: usize) -> Vec<usize> {
if n < 7 {
return [2, 3, 5]
.iter()
.filter_map(|&x| (x <= n).then_some_(x))
.collect();
}
let nf = n as f64;
let mut primes = Vec::with_capacity((nf / nf.ln() * 1.2).floor() as usize);
primes.push(2);
primes.push(3);
primes.push(5);
let mut unmarked_numbers = vec![true; n + 1];
// Wheel factorization
let mut p = 7 - INC.last().unwrap();
for i in INC.len() - 1.. {
p += INC[i % INC.len()];
if p > n {
break;
}
if !unmarked_numbers[p] {
continue;
}
primes.push(p);
for px in (p * p..=n).step_by(p) {
unmarked_numbers[px] = false;
}
}
primes
}
pub trait IterExt: Iterator {
fn easy_join(&mut self, separator: &str) -> String
where
Self::Item: Display,
{
self.map(|i| format!("{}", i))
.collect::<Vec<_>>()
.join(separator)
}
fn group_with(self) -> HashMap<Self::Item, Vec<Self::Item>>
where
Self: Sized,
Self::Item: Eq + Hash + Clone,
{
self.group_with_key(|x| x.clone())
}
fn group_with_key<K, F>(self, mut f: F) -> HashMap<K, Vec<Self::Item>>
where
Self: Sized,
F: FnMut(&Self::Item) -> K,
K: Eq + Hash,
{
self.fold(HashMap::new(), |mut acc, item| {
acc.entry(f(&item)).or_insert(vec![]).push(item);
acc
})
}
}
impl<TItem, TTrait> IterExt for TTrait where TTrait: Iterator<Item = TItem> {}
pub trait VecExt<T> {
fn add_like_string(&mut self) -> T;
}
impl<T> VecExt<T> for Vec<T>
where
T: GenericInteger,
{
#[inline]
fn add_like_string(&mut self) -> T {
if let Ok(value) = self.iter().easy_join("").parse::<T>() {
value
} else {
panic!("Invalid value")
}
}
}
#[allow(unused_macros)]
#[macro_export]
macro_rules! swap {
($v1:expr, $v2:expr) => {
let buf = $v1;
$v1 = $v2;
$v2 = buf;
};
}
#[macro_export]
macro_rules! invert_index {
($v:expr) => {{
let mut goal = vec![0usize; $v.len()];
for (i, v) in $v.iter().enumerate() {
goal[*v] = i;
}
goal
}};
}
pub trait Then: Into<bool> {
fn then_<T, F>(self, f: F) -> Option<T>
where
F: Fn() -> T,
{
if self.into() {
Some(f())
} else {
None
}
}
}
impl Then for bool {}
pub trait ThenSome: Into<bool> {
fn then_some_<T>(self, t: T) -> Option<T> {
if self.into() {
Some(t)
} else {
None
}
}
}
impl ThenSome for bool {}
// From https://kuretchi.hateblo.jp/entry/rust_nested_vec
#[allow(unused_macros)]
macro_rules! nested_vec {
($e:expr; $n:expr) => {
vec![$e; $n]
};
($e:expr; $n:expr $(; $m:expr)+) => {
vec![nested_vec!($e $(; $m)+); $n]
};
}
// From https://maguro.dev/debug-macro/ with some modification
#[allow(unused_macros)]
macro_rules! dbg {
() => {
#[cfg(debug_assertions)]
eprintln!();
};
($($a:expr),* $(,)*) => {
#[cfg(debug_assertions)]
eprintln!(concat!($("| ", stringify!($a), "={:?} "),*, "|"), $(&$a),*);
};
}
#[allow(unused_macros)]
macro_rules! dbg_raw {
() => {
#[cfg(debug_assertions)]
eprintln!();
};
($a:expr) => {
#[cfg(debug_assertions)]
eprintln!("{:?}", $a);
};
($a:expr, $($b:expr),+ $(,)*) => {
#[cfg(debug_assertions)]
eprint!("{:?}", $a);
dbg_raw!($($b),+);
};
}
// From https://qiita.com/hatoo@github/items/fa14ad36a1b568d14f3e
#[derive(PartialEq, PartialOrd)]
struct Total<T>(T);
impl<T: PartialEq> Eq for Total<T> {}
impl<T: PartialOrd> Ord for Total<T> {
fn cmp(&self, other: &Total<T>) -> Ordering {
self.0.partial_cmp(&other.0).unwrap()
}
}
pub fn index_to_ascii_gen(base: char) -> impl Fn(usize) -> char {
move |index| (index as u8 + base as u8) as char
}
pub fn ascii_to_index_gen(base: char) -> impl Fn(char) -> usize {
move |ascii| ascii as usize - base as usize
}
mod point {
use super::GenericInteger;
use std::ops::*;
#[derive(Copy, Clone, Eq, PartialEq, Hash, Debug, Default)]
pub struct Point2d<T> {
pub x: T,
pub y: T,
}
impl<T> Point2d<T>
where
T: GenericInteger,
{
#[inline]
pub fn zero() -> Self {
Self {
x: T::zero(),
y: T::zero(),
}
}
#[inline]
pub fn one() -> Self {
Self {
x: T::one(),
y: T::one(),
}
}
}
impl<T> Point2d<T> {
pub fn cast_to<U>(self) -> Point2d<U>
where
U: From<T>,
{
Point2d {
x: U::from(self.x),
y: U::from(self.y),
}
}
pub fn dot(&self, rhs: &Self) -> T
where
T: Add<Output = T> + Mul<Output = T> + Clone,
{
self.x.clone() * rhs.x.clone() + self.y.clone() * rhs.y.clone()
}
}
impl Point2d<usize> {
pub fn to_f64(&self) -> Point2d<f64> {
Point2d {
x: self.x as f64,
y: self.y as f64,
}
}
}
impl Point2d<f64> {
pub fn length(&self) -> f64 {
self.x.hypot(self.y)
}
}
impl<T> Add for Point2d<T>
where
T: Add<Output = T>,
{
type Output = Self;
fn add(self, rhs: Self) -> Self::Output {
Self {
x: self.x + rhs.x,
y: self.y + rhs.y,
}
}
}
impl<T> AddAssign for Point2d<T>
where
T: AddAssign,
{
fn add_assign(&mut self, rhs: Self) {
self.x += rhs.x;
self.y += rhs.y;
}
}
impl<T> Sub for Point2d<T>
where
T: Sub<Output = T>,
{
type Output = Self;
fn sub(self, rhs: Self) -> Self::Output {
Self {
x: self.x - rhs.x,
y: self.y - rhs.y,
}
}
}
impl<T> SubAssign for Point2d<T>
where
T: SubAssign,
{
fn sub_assign(&mut self, rhs: Self) {
self.x -= rhs.x;
self.y -= rhs.y;
}
}
impl<T> Mul<T> for Point2d<T>
where
T: Mul<Output = T> + Clone,
{
type Output = Self;
fn mul(self, rhs: T) -> Self::Output {
Self {
x: self.x * rhs.clone(),
y: self.y * rhs,
}
}
}
impl<T> MulAssign<T> for Point2d<T>
where
T: MulAssign + Clone,
{
fn mul_assign(&mut self, rhs: T) {
self.x *= rhs.clone();
self.y *= rhs;
}
}
impl<T> Div<T> for Point2d<T>
where
T: Div<Output = T> + Clone,
{
type Output = Self;
fn div(self, rhs: T) -> Self::Output {
Self {
x: self.x / rhs.clone(),
y: self.y / rhs,
}
}
}
impl<T> DivAssign<T> for Point2d<T>
where
T: DivAssign + Clone,
{
fn div_assign(&mut self, rhs: T) {
self.x /= rhs.clone();
self.y /= rhs;
}
}
impl<T> Rem<T> for Point2d<T>
where
T: Rem<Output = T> + Clone,
{
type Output = Self;
fn rem(self, rhs: T) -> Self::Output {
Self {
x: self.x % rhs.clone(),
y: self.y % rhs,
}
}
}
impl<T> RemAssign<T> for Point2d<T>
where
T: RemAssign + Clone,
{
fn rem_assign(&mut self, rhs: T) {
self.x %= rhs.clone();
self.y %= rhs;
}
}
impl<T> Neg for Point2d<T>
where
T: Neg<Output = T>,
{
type Output = Self;
fn neg(self) -> Self::Output {
Self {
x: -self.x,
y: -self.y,
}
}
}
}
pub fn div_ceil<T: GenericInteger>(dividend: T, divisor: T) -> T {
let rounded_towards_zero_quotient = dividend / divisor;
let divided_evenly = (dividend % divisor) == T::zero();
if divided_evenly {
return rounded_towards_zero_quotient;
}
let was_rounded_down = (divisor > T::zero()) == (dividend > T::zero());
if was_rounded_down {
rounded_towards_zero_quotient + T::one()
} else {
rounded_towards_zero_quotient
}
}
mod multiset {
use std::borrow::Borrow;
use std::collections::*;
use std::fmt;
use std::hash::Hash;
use std::iter::FromIterator;
use std::ops::RangeBounds;
pub struct HashMultiset<T> {
map: HashMap<T, usize>,
}
impl<T> HashMultiset<T> {
#[inline]
pub fn capacity(&self) -> usize {
self.map.capacity()
}
#[inline]
pub fn len(&self) -> usize {
self.map.values().sum()
}
#[inline]
pub fn is_empty(&self) -> bool {
self.map.is_empty()
}
#[inline]
pub fn clear(&mut self) {
self.map.clear()
}
}
impl<T> HashMultiset<T>
where
T: Eq + Hash,
{
#[inline]
pub fn new() -> Self {
Self {
map: HashMap::new(),
}
}
#[inline]
pub fn with_capacity(capacity: usize) -> Self {
Self {
map: HashMap::with_capacity(capacity),
}
}
#[inline]
pub fn value_quantity_pairs(&self) -> hash_map::Iter<'_, T, usize> {
self.map.iter()
}
#[inline]
pub fn contains<Q: ?Sized>(&self, value: &Q) -> bool
where
T: Borrow<Q>,
Q: Hash + Eq,
{
self.map.contains_key(&value)
}
#[inline]
pub fn get<Q: ?Sized>(&self, value: &Q) -> Option<&T>
where
T: Borrow<Q>,
Q: Hash + Eq,
{
self.map.get_key_value(value).map(|(k, _)| k)
}
#[inline]
pub fn insert(&mut self, value: T) -> bool {
*self.map.entry(value).or_insert(0) += 1;
true
}
pub fn remove_single<Q: ?Sized>(&mut self, value: &Q) -> bool
where
T: Borrow<Q>,
Q: Hash + Eq,
{
if let Some(v) = self.map.get_mut(&value) {
*v -= 1;
if *v == 0 {
self.map.remove(&value);
}
true
} else {
false
}
}
#[inline]
pub fn remove_many<Q: ?Sized>(&mut self, value: &Q) -> bool
where
T: Borrow<Q>,
Q: Hash + Eq,
{
self.map.remove(value).is_some()
}
}
impl<T> PartialEq for HashMultiset<T>
where
T: Eq + Hash,
{
fn eq(&self, other: &Self) -> bool {
self.map.eq(&other.map)
}
}
impl<T> Eq for HashMultiset<T> where T: Eq + Hash {}
impl<T> fmt::Debug for HashMultiset<T>
where
T: Eq + Hash + fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_set().entries(self.map.iter()).finish()
}
}
impl<T> FromIterator<T> for HashMultiset<T>
where
T: Eq + Hash,
{
#[inline]
fn from_iter<I: IntoIterator<Item = T>>(iter: I) -> HashMultiset<T> {
let mut set = Self::new();
set.extend(iter);
set
}
}
impl<T> Extend<T> for HashMultiset<T>
where
T: Eq + Hash,
{
#[inline]
fn extend<I: IntoIterator<Item = T>>(&mut self, iter: I) {
for k in iter.into_iter() {
self.insert(k);
}
}
}
impl<'a, T> Extend<&'a T> for HashMultiset<T>
where
T: 'a + Eq + Hash + Copy,
{
#[inline]
fn extend<I: IntoIterator<Item = &'a T>>(&mut self, iter: I) {
self.extend(iter.into_iter().cloned());
}
}
impl<T> Default for HashMultiset<T>
where
T: Eq + Hash,
{
#[inline]
fn default() -> Self {
Self {
map: HashMap::default(),
}
}
}
pub struct BTreeMultiset<T> {
map: BTreeMap<T, usize>,
}
impl<T> BTreeMultiset<T> {
#[inline]
pub fn len(&self) -> usize {
self.map.values().sum()
}
#[inline]
pub fn is_empty(&self) -> bool {
self.map.is_empty()
}
}
impl<T: Ord> BTreeMultiset<T> {
#[inline]
pub fn new() -> Self {
Self {
map: BTreeMap::new(),
}
}
#[inline]
pub fn value_quantity_pairs(&self) -> btree_map::Iter<'_, T, usize> {
self.map.iter()
}
#[inline]
pub fn key_range<K: ?Sized, R>(&self, range: R) -> btree_map::Range<'_, T, usize>
where
K: Ord,
T: Borrow<K>,
R: RangeBounds<K>,
{
self.map.range(range)
}
#[inline]
pub fn clear(&mut self) {
self.map.clear()
}
#[inline]
pub fn contains<Q: ?Sized>(&self, value: &Q) -> bool
where
T: Borrow<Q>,
Q: Ord,
{
self.map.contains_key(value)
}
pub fn insert(&mut self, value: T) -> bool {
*self.map.entry(value).or_insert(0) += 1;
true
}
pub fn remove_single<Q: ?Sized>(&mut self, value: &Q) -> bool
where
T: Borrow<Q>,
Q: Ord,
{
if let Some(v) = self.map.get_mut(&value) {
*v -= 1;
if *v == 0 {
self.map.remove(&value);
}
true
} else {
false
}
}
#[inline]
pub fn remove_many<Q: ?Sized>(&mut self, value: &Q) -> bool
where
T: Borrow<Q>,
Q: Ord,
{
self.map.remove(value).is_some()
}
}
impl<T: fmt::Debug> fmt::Debug for BTreeMultiset<T> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_set().entries(self.map.iter()).finish()
}
}
impl<T: Ord> FromIterator<T> for BTreeMultiset<T> {
fn from_iter<I: IntoIterator<Item = T>>(iter: I) -> Self {
let mut set = BTreeMultiset::new();
set.extend(iter);
set
}
}
impl<T: Ord> Extend<T> for BTreeMultiset<T> {
#[inline]
fn extend<Iter: IntoIterator<Item = T>>(&mut self, iter: Iter) {
iter.into_iter().for_each(move |elem| {
self.insert(elem);
});
}
}
impl<'a, T: 'a + Ord + Copy> Extend<&'a T> for BTreeMultiset<T> {
#[inline]
fn extend<I: IntoIterator<Item = &'a T>>(&mut self, iter: I) {
self.extend(iter.into_iter().cloned());
}
}
impl<T: Ord> Default for BTreeMultiset<T> {
fn default() -> Self {
Self::new()
}
}
}
// -- end of helpers
fn main() {
use std::io::*;
let stdio = stdin();
let input = stdio.lock();
let mut stdout = stdout();
let output = BufWriter::new(stdout.lock());
process(input, output).expect("Should not emit error");
stdout.flush().unwrap();
}
#[allow(non_snake_case, unused_mut, unused_variables)]
fn process<R, W>(reader: R, mut writer: W) -> std::io::Result<()>
where
R: std::io::BufRead,
W: std::io::Write,
{
let mut _handler = prepare_input! { stdin = reader };
#[allow(unused_macros)]
macro_rules! input {
($($r:tt)*) => {
input_original! { handler = _handler; $($r)* }
};
}
#[allow(unused_macros)]
macro_rules! print {
($($arg:tt)*) => {
write!(writer, $($arg)*)?;
}
}
#[allow(unused_macros)]
macro_rules! println {
() => {
writeln!(writer)?;
};
($($arg:tt)*) => {
writeln!(writer, $($arg)*)?;
}
}
{
input! {
n: usize, a: usize1, b: usize1,
p: usize1, q: usize1, r: usize1, s: usize1,
}
let mut rows = nested_vec![false; q - p + 1; s - r + 1];
for i in p..=q {
for j in r..=s {
// if a + k == i && b + k == j => k == k {
if i + b == j + a {
rows[i - p][j - r] = true;
}
// if a + k == i && b - k == j => k == k {
// if i - a == b - j {
if i + j == a + b {
rows[i - p][j - r] = true;
}
}
}
for row in rows.iter() {
for cell in row.iter() {
print!("{}", if *cell { "#" } else { "." });
}
println!();
}
}
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn sample1() {
assert_judge!(
process,
"
5 3 2
1 5 1 5
",
"
...#.
#.#..
.#...
#.#..
...#.
"
);
}
#[test]
fn sample2() {
assert_judge!(
process,
"
5 3 3
4 5 2 5
",
"
#.#.
...#
"
);
}
#[test]
fn sample3() {
assert_judge!(
process,
"
1000000000000000000 999999999999999999 999999999999999999
999999999999999998 1000000000000000000 999999999999999998 1000000000000000000
",
"
#.#
.#.
#.#
"
);
}
}
Submission Info
| Submission Time |
|
| Task |
C - X drawing |
| User |
occar421 |
| Language |
Rust (1.42.0) |
| Score |
300 |
| Code Size |
32748 Byte |
| Status |
AC |
| Exec Time |
53 ms |
| Memory |
18412 KiB |
Judge Result
| Set Name |
Sample |
All |
| Score / Max Score |
0 / 0 |
300 / 300 |
| Status |
|
|
| Set Name |
Test Cases |
| Sample |
example_00.txt, example_01.txt, example_02.txt |
| All |
example_00.txt, example_01.txt, example_02.txt, hand_00.txt, hand_01.txt, hand_02.txt, hand_03.txt, hand_04.txt, hand_05.txt, hand_06.txt, hand_07.txt, hand_08.txt, hand_09.txt, hand_10.txt, hand_11.txt, random_00.txt, random_01.txt, random_02.txt, random_03.txt, random_04.txt, random_05.txt, random_06.txt, random_07.txt, random_08.txt, random_09.txt, random_10.txt, random_11.txt, random_12.txt, random_13.txt, random_14.txt, random_15.txt, random_16.txt, random_17.txt, random_18.txt, random_19.txt |
| Case Name |
Status |
Exec Time |
Memory |
| example_00.txt |
AC |
10 ms |
2060 KiB |
| example_01.txt |
AC |
2 ms |
2116 KiB |
| example_02.txt |
AC |
2 ms |
2112 KiB |
| hand_00.txt |
AC |
18 ms |
2188 KiB |
| hand_01.txt |
AC |
20 ms |
2100 KiB |
| hand_02.txt |
AC |
53 ms |
18412 KiB |
| hand_03.txt |
AC |
13 ms |
2196 KiB |
| hand_04.txt |
AC |
16 ms |
2220 KiB |
| hand_05.txt |
AC |
1 ms |
1976 KiB |
| hand_06.txt |
AC |
8 ms |
2248 KiB |
| hand_07.txt |
AC |
9 ms |
2168 KiB |
| hand_08.txt |
AC |
13 ms |
2240 KiB |
| hand_09.txt |
AC |
24 ms |
2148 KiB |
| hand_10.txt |
AC |
20 ms |
2132 KiB |
| hand_11.txt |
AC |
26 ms |
2036 KiB |
| random_00.txt |
AC |
16 ms |
2264 KiB |
| random_01.txt |
AC |
16 ms |
2348 KiB |
| random_02.txt |
AC |
15 ms |
2280 KiB |
| random_03.txt |
AC |
20 ms |
2208 KiB |
| random_04.txt |
AC |
14 ms |
2356 KiB |
| random_05.txt |
AC |
17 ms |
2312 KiB |
| random_06.txt |
AC |
13 ms |
2252 KiB |
| random_07.txt |
AC |
10 ms |
2184 KiB |
| random_08.txt |
AC |
19 ms |
2212 KiB |
| random_09.txt |
AC |
10 ms |
2224 KiB |
| random_10.txt |
AC |
15 ms |
2352 KiB |
| random_11.txt |
AC |
6 ms |
2164 KiB |
| random_12.txt |
AC |
13 ms |
2348 KiB |
| random_13.txt |
AC |
10 ms |
2360 KiB |
| random_14.txt |
AC |
20 ms |
2264 KiB |
| random_15.txt |
AC |
6 ms |
2328 KiB |
| random_16.txt |
AC |
12 ms |
2160 KiB |
| random_17.txt |
AC |
10 ms |
2396 KiB |
| random_18.txt |
AC |
14 ms |
2496 KiB |
| random_19.txt |
AC |
10 ms |
2344 KiB |