Skip to content

Commit

Permalink
Add atcoder/arc176/b.rs atcoder/arc176/c.rs atcoder/arc176/d-lib.rs a…
Browse files Browse the repository at this point in the history
…tcoder/arc176/d.rs atcoder/arc176/e.rs atcoder/arc176/remain.txt
  • Loading branch information
koba-e964 committed Jun 18, 2024
1 parent d4ee856 commit 570b83e
Show file tree
Hide file tree
Showing 6 changed files with 963 additions and 0 deletions.
44 changes: 44 additions & 0 deletions atcoder/arc176/b.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
use std::io::Read;

fn get_word() -> String {
let stdin = std::io::stdin();
let mut stdin=stdin.lock();
let mut u8b: [u8; 1] = [0];
loop {
let mut buf: Vec<u8> = Vec::with_capacity(16);
loop {
let res = stdin.read(&mut u8b);
if res.unwrap_or(0) == 0 || u8b[0] <= b' ' {
break;
} else {
buf.push(u8b[0]);
}
}
if buf.len() >= 1 {
let ret = String::from_utf8(buf).unwrap();
return ret;
}
}
}

fn get<T: std::str::FromStr>() -> T { get_word().parse().ok().unwrap() }

fn main() {
let t: i32 = get();
for _ in 0..t {
let mut n: i64 = get();
let m: i64 = get();
let k: i64 = get();
if n >= k {
n -= k;
if m - k > 1 {
n %= m - k;
n += k;
} else {
println!("0");
continue;
}
}
println!("{}", [6, 2, 4, 8][(n % 4) as usize]);
}
}
113 changes: 113 additions & 0 deletions atcoder/arc176/c.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,113 @@
// https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
macro_rules! input {
($($r:tt)*) => {
let stdin = std::io::stdin();
let mut bytes = std::io::Read::bytes(std::io::BufReader::new(stdin.lock()));
let mut next = move || -> String{
bytes.by_ref().map(|r|r.unwrap() as char)
.skip_while(|c|c.is_whitespace())
.take_while(|c|!c.is_whitespace())
.collect()
};
input_inner!{next, $($r)*}
};
}

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)*}
};
}

macro_rules! read_value {
($next:expr, ( $($t:tt),* )) => { ($(read_value!($next, $t)),*) };
($next:expr, [ $t:tt ; $len:expr ]) => {
(0..$len).map(|_| read_value!($next, $t)).collect::<Vec<_>>()
};
($next:expr, usize1) => (read_value!($next, usize) - 1);
($next:expr, $t:ty) => ($next().parse::<$t>().expect("Parse error"));
}

fn main() {
input! {
n: usize, m: usize,
abc: [(usize1, usize1, usize1); m],
}
let mut ev = vec![vec![]; n];
for (a, b, c) in abc {
ev[c].push((a, b));
}
const MOD: i64 = 998_244_353;
let mut ans = 1;
let mut last = 0;
let mut dec = vec![false; n];
for i in 0..n {
if ev[i].is_empty() {
continue;
}
let get = |j: usize| {
let mut v = vec![];
let (k1, k2) = ev[i][j];
if !dec[k1] {
v.push(k1);
}
if !dec[k2] {
v.push(k2);
}
v
};
let mut int = get(0);
let mut uni = get(0);
for j in 1..ev[i].len() {
let mut new_int = vec![];
let cur = get(j);
for &x in &int {
if cur.contains(&x) {
new_int.push(x);
}
}
int = new_int;
uni.extend_from_slice(&cur);
}
if int.is_empty() {
println!("0");
return;
}
if int.len() == 2 {
assert_eq!(uni.len(), 2);
for &v in &uni {
assert!(!dec[v]);
dec[v] = true;
}
if last >= i {
ans = 0;
continue;
}
ans = ans * (i - last) as i64 % MOD * 2 % MOD;
last += 2;
continue;
}
assert_eq!(int.len(), 1);
uni.sort(); uni.dedup();
for &v in &uni {
assert!(!dec[v]);
dec[v] = true;
}
// *= P(i - last, uni.len() - 1)
if last + uni.len() - 1 > i {
ans = 0;
continue;
}
for j in 1..uni.len() {
ans = ans * (i + 1 - last - j) as i64 % MOD;
}
last += uni.len();
}
for i in last..n {
ans = ans * (n - i) as i64 % MOD;
}
println!("{}", ans);
}
Loading

0 comments on commit 570b83e

Please sign in to comment.