complete 4779.rs 32720.rs

This commit is contained in:
2026-03-23 18:26:12 +09:00
parent 4e380a44ac
commit f81952bb9f
2 changed files with 107 additions and 0 deletions

View File

@@ -0,0 +1,75 @@
use std::io::{read_to_string, stdin};
const MOD: usize = 1_000_000_007;
const N_MAX: usize = 1_000_000;
fn get_mod_power(n: usize, p: usize) -> usize {
let mut x = p;
let mut curr = n % MOD;
let mut res = 1;
while x > 0 {
if x & 1 == 1 {
res *= curr;
res %= MOD;
}
x >>= 1;
curr *= curr;
curr %= MOD;
}
res
}
fn main() {
let temp = read_to_string(stdin()).unwrap();
let mut iter = temp
.split_ascii_whitespace()
.map(|x| x.parse::<usize>().unwrap());
let (n, k) = (iter.next().unwrap(), iter.next().unwrap());
let mut fac = vec![0; N_MAX + 1];
let mut facinv = vec![0; N_MAX + 1];
fac[0] = 1;
facinv[0] = 1;
fac[1] = 1;
facinv[1] = 1;
for i in 2..=N_MAX {
fac[i] = (fac[i - 1] * i) % MOD;
}
facinv[N_MAX] = get_mod_power(fac[N_MAX], MOD - 2);
for i in (2..N_MAX).rev() {
facinv[i] = facinv[i + 1] * (i + 1) % MOD;
}
let get_nck = |n: usize, k: usize| (fac[n] * facinv[k] % MOD) * facinv[n - k] % MOD;
//
let mut res = 1;
let mut total = n;
let mut vis = vec![false; n + 1];
if k == 1 {
res = 1;
} else {
for i in 1..=n {
if i + k > n {
break;
}
let mut cnt = 0;
for j in (i..=n).step_by(k) {
if vis[j] {
continue;
}
vis[j] = true;
cnt += 1;
}
res *= get_nck(total, cnt);
total -= cnt;
res %= MOD;
}
if total > 1 {
res *= fac[total];
res %= MOD;
}
}
println!("{}", res);
}

View File

@@ -0,0 +1,32 @@
use std::io::{read_to_string, stdin};
fn cantor_string(line: &mut Vec<char>, lo: usize, n: usize) {
if n == 0 {
return;
}
let x = 3usize.pow(n as u32 - 1);
for i in lo + x..lo + 2 * x {
line[i] = ' ';
}
cantor_string(line, lo, n - 1);
cantor_string(line, lo + 2 * x, n - 1);
}
fn main() {
let temp = read_to_string(stdin()).unwrap();
let mut iter = temp
.split_ascii_whitespace()
.map(|x| x.parse::<usize>().unwrap());
loop {
let n = match iter.next() {
Some(n) => n,
None => {
break;
}
};
let x = 3usize.pow(n as u32);
let mut line = vec!['-'; x];
cantor_string(&mut line, 0, n);
println!("{}", line.iter().collect::<String>());
}
}