結果
問題 | No.1966 Median of Divisors |
ユーザー |
![]() |
提出日時 | 2022-06-03 22:52:22 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 257 ms / 2,000 ms |
コード長 | 13,739 bytes |
コンパイル時間 | 23,579 ms |
コンパイル使用メモリ | 399,572 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-21 03:08:25 |
合計ジャッジ時間 | 15,607 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 12 |
コンパイルメッセージ
warning: unnecessary parentheses around type --> src/main.rs:74:15 | 74 | fn readi() -> (i64) { | ^ ^ | = note: `#[warn(unused_parens)]` on by default help: remove these parentheses | 74 - fn readi() -> (i64) { 74 + fn readi() -> i64 { | warning: unnecessary parentheses around assigned value --> src/main.rs:302:19 | 302 | self.0 *= (rhs.0 % MOD); | ^ ^ | help: remove these parentheses | 302 - self.0 *= (rhs.0 % MOD); 302 + self.0 *= rhs.0 % MOD; | warning: use of deprecated function `std::intrinsics::drop_in_place`: no longer an intrinsic - use `ptr::drop_in_place` directly --> src/main.rs:17:22 | 17 | use std::intrinsics::drop_in_place; | ^^^^^^^^^^^^^ | = note: `#[warn(deprecated)]` on by default warning: unused variable: `a2` --> src/main.rs:453:18 | 453 | let (a2, e2) = self.mod_fact(k, p); | ^^ help: if this is intentional, prefix it with an underscore: `_a2` | = note: `#[warn(unused_variables)]` on by default warning: unused variable: `moi` --> src/main.rs:458:21 | 458 | let moi = self.mod_inverse(a3 % p, p); | ^^^ help: if this is intentional, prefix it with an underscore: `_moi` warning: unused variable: `a2` --> src/main.rs:478:18 | 478 | let (a2, e2) = self.mod_fact(k, p); | ^^ help: if this is intentional, prefix it with an underscore: `_a2` warning: unused variable: `a3` --> src/main.rs:479:18 | 479 | let (a3, e3) = self.mod_fact(n - k, p); | ^^ help: if this is intentional, prefix it with an underscore: `_a3` warning: unused variable: `i` --> src/main.rs:523:9 | 523 | for i in 0..m { | ^ help: if this is intentional, prefix it with an underscore: `_i` warning: variable `su` is assigned to, but never used
ソースコード
// -*- coding:utf-8-unix -*-// #![feature(map_first_last)]#![allow(dead_code)]#![allow(unused_imports)]#![allow(unused_macros)]use std::any::Any;use std::cmp::Ordering::*;use std::collections::*;use std::convert::*;use std::convert::{From, Into};use std::error::Error;use std::f64::consts::PI;use std::fmt::Debug;use std::fmt::Display;use std::fs::File;use std::hash::Hash;use std::intrinsics::drop_in_place;use std::io::prelude::*;use std::io::*;use std::iter::Filter;use std::marker::Copy;use std::mem::*;use std::ops::Bound::*;use std::ops::RangeBounds;use std::ops::{Add, Mul, Neg, Sub};use std::process;use std::slice::from_raw_parts;use std::str;use std::vec;const INF: i64 = 1223372036854775807;const UINF: usize = INF as usize;// const FINF: f64 = 122337203685.0;const LINF: i64 = 2147483647;const FINF: f64 = LINF as f64;const INF128: i128 = 1223372036854775807000000000000;const MOD: i64 = 1000000007;// const MOD: i64 = 998244353;const UMOD: usize = MOD as usize;const MPI: f64 = 3.14159265358979323846264338327950288f64;// const MOD: i64 = INF;use std::cmp::*;use std::collections::*;use std::io::stdin;use std::io::stdout;use std::io::Write;macro_rules! p {($x:expr) => {println!("{}", $x);};}macro_rules! d {($x:expr) => {println!("{:?}", $x);};}fn main() {solve();}// use str::Chars;#[allow(dead_code)]fn read<T: std::str::FromStr>() -> T {let mut s = String::new();std::io::stdin().read_line(&mut s).ok();s.trim().parse().ok().unwrap()}#[allow(dead_code)]fn readi() -> (i64) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();iter.next().unwrap().parse::<i64>().unwrap()}#[allow(dead_code)]fn read_vec<T: std::str::FromStr>() -> Vec<T> {read::<String>().split_whitespace().map(|e| e.parse().ok().unwrap()).collect()}#[allow(dead_code)]fn read_mat<T: std::str::FromStr>(n: u32) -> Vec<Vec<T>> {(0..n).map(|_| read_vec()).collect()}#[allow(dead_code)]fn readii() -> (i64, i64) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<i64>().unwrap(),iter.next().unwrap().parse::<i64>().unwrap(),)}fn readff() -> (f64, f64) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<f64>().unwrap(),iter.next().unwrap().parse::<f64>().unwrap(),)}#[allow(dead_code)]fn readiii() -> (i64, i64, i64) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<i64>().unwrap(),iter.next().unwrap().parse::<i64>().unwrap(),iter.next().unwrap().parse::<i64>().unwrap(),)}#[allow(dead_code)]fn readuu() -> (usize, usize) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<usize>().unwrap(),iter.next().unwrap().parse::<usize>().unwrap(),)}fn readcc() -> (char, char) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<char>().unwrap(),iter.next().unwrap().parse::<char>().unwrap(),)}fn readuuu() -> (usize, usize, usize) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<usize>().unwrap(),iter.next().unwrap().parse::<usize>().unwrap(),iter.next().unwrap().parse::<usize>().unwrap(),)}#[allow(dead_code)]fn readiiii() -> (i64, i64, i64, i64) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<i64>().unwrap(),iter.next().unwrap().parse::<i64>().unwrap(),iter.next().unwrap().parse::<i64>().unwrap(),iter.next().unwrap().parse::<i64>().unwrap(),)}#[allow(dead_code)]fn readuuuu() -> (usize, usize, usize, usize) {let mut str = String::new();let _ = stdin().read_line(&mut str).unwrap();let mut iter = str.split_whitespace();(iter.next().unwrap().parse::<usize>().unwrap(),iter.next().unwrap().parse::<usize>().unwrap(),iter.next().unwrap().parse::<usize>().unwrap(),iter.next().unwrap().parse::<usize>().unwrap(),)}fn divisors(n: usize) -> Vec<usize> {let mut divisors = Vec::new();// n := i * x とおくと、 i が i > root(n) の時、 i はすでに ある x に探索されているから// i <= root(n) まで探索すればよいfor i in 1..=(f64::sqrt(n as f64) + 1e-9) as usize {// i で n が割り切れた場合if n % i == 0 {// 約数リストに格納divisors.push(i);// n := i * x の x を格納。ただし x := i の時は除くif i != n / i {divisors.push(n / i);}}}divisors.sort();divisors}macro_rules! M {(a :expr ) => {M::new({ a })};}#[derive(Copy, Clone, Debug)]pub struct M(i64);impl M {fn new(x: i64) -> Self {M(x.rem_euclid(MOD))}fn pow(self, n: usize) -> Self {match n {0 => M::new(1),_ => {let mut a = self.pow(n >> 1);a *= a;if n & 1 == 1 {a *= self;}a}}}fn inv(self) -> Self {self.pow((MOD - 2) as usize)}}impl std::ops::Neg for M {type Output = M;fn neg(self) -> Self::Output {Self::new(-self.0)}}impl std::ops::AddAssign<M> for M {fn add_assign(&mut self, rhs: Self) {self.0 += rhs.0;self.0 %= MOD;}}impl std::ops::AddAssign<i64> for M {fn add_assign(&mut self, rhs: i64) {*self += M::new(rhs);}}impl std::ops::AddAssign<usize> for M {fn add_assign(&mut self, rhs: usize) {*self += M::new(rhs as i64);}}impl<T> std::ops::Add<T> for MwhereM: std::ops::AddAssign<T>,{type Output = Self;fn add(self, other: T) -> Self {let mut res = self;res += other;res}}impl std::ops::SubAssign<M> for M {fn sub_assign(&mut self, rhs: Self) {self.0 -= rhs.0;if self.0 < 0 {self.0 += MOD;}}}impl std::ops::SubAssign<i64> for M {fn sub_assign(&mut self, rhs: i64) {*self -= M::new(rhs);if (*self).0 < 0 {self.0 += MOD;}}}impl std::ops::SubAssign<usize> for M {fn sub_assign(&mut self, rhs: usize) {*self -= M::new(rhs as i64);if (*self).0 < 0 {self.0 += MOD;}}}impl<T> std::ops::Sub<T> for MwhereM: std::ops::SubAssign<T>,{type Output = Self;fn sub(self, other: T) -> Self {let mut res = self;res -= other;res}}impl std::ops::MulAssign<M> for M {fn mul_assign(&mut self, rhs: Self) {self.0 %= MOD;self.0 *= (rhs.0 % MOD);self.0 %= MOD;}}impl std::ops::MulAssign<i64> for M {fn mul_assign(&mut self, rhs: i64) {*self *= M::new(rhs);}}impl std::ops::MulAssign<usize> for M {fn mul_assign(&mut self, rhs: usize) {*self *= M::new(rhs as i64);}}impl<T> std::ops::Mul<T> for MwhereM: std::ops::MulAssign<T>,{type Output = Self;fn mul(self, other: T) -> Self {let mut res = self;res *= other;res}}impl std::ops::DivAssign<M> for M {fn div_assign(&mut self, rhs: Self) {*self *= rhs.inv();}}impl std::ops::DivAssign<i64> for M {fn div_assign(&mut self, rhs: i64) {*self /= M::new(rhs);}}impl std::ops::DivAssign<usize> for M {fn div_assign(&mut self, rhs: usize) {*self /= M::new(rhs as i64);}}impl<T> std::ops::Div<T> for MwhereM: std::ops::DivAssign<T>,{type Output = Self;fn div(self, other: T) -> Self {let mut res = self;res /= other;res}}impl std::fmt::Display for M {fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {write!(f, "{}", self.0)}}impl std::ops::Deref for M {type Target = i64;fn deref(&self) -> &Self::Target {&self.0}}impl std::ops::DerefMut for M {fn deref_mut(&mut self) -> &mut Self::Target {&mut self.0}}#[allow(dead_code)]pub fn gcd(a: usize, b: usize) -> usize {if b == 0 {a} else {gcd(b, a % b)}}#[allow(dead_code)]pub fn lcm(a: usize, b: usize) -> usize {a / gcd(a, b) * b}#[allow(dead_code)]/// (gcd, x, y)pub fn extgcd(a: i64, b: i64) -> (i64, i64, i64) {if b == 0 {(a, 1, 0)} else {let (gcd, x, y) = extgcd(b, a % b);(gcd, y, x - (a / b) * y)}}#[allow(dead_code)]/// x ^ n % mpub fn mod_pow(x: usize, n: usize, m: usize) -> usize {let mut res = 1;let mut x = x % m;let mut n = n;while n > 0 {if n & 1 == 1 {res = (res * x) % m;}x = (x * x) % m;n >>= 1;}res}pub struct Combination {m: usize,f_table: Vec<usize>,moi: Vec<usize>,}impl Combination {// 0 <= size <= 10^8 is constrained.pub fn new(mod_num: usize, table_size: usize) -> Self {Self {m: mod_num,f_table: vec![0; table_size],moi: vec![0; 0],}}pub fn build(&mut self) {let size = self.f_table.len();self.f_table = self.fact_table(size, self.m);self.moi = self.fact_inv_table(size, self.m);}fn fact_table(&mut self, len: usize, m: usize) -> Vec<usize> {let mut res = vec![1; len + 1];for i in 1..len + 1 {res[i] = (i as usize * res[i - 1]) % m;}res}fn fact_inv_table(&mut self, len: usize, m: usize) -> Vec<usize> {let mut res = vec![1; len + 1];for i in 1..len + 1 {res[i] = (res[i - 1] * self.mod_inverse(i, m)) % m;}res}pub fn p(&mut self, n: usize, k: usize) -> i64 {let p = MOD as usize;if k == 0 {return 1;}if n < k {0} else {let (a1, e1) = self.mod_fact(n, p);let (a2, e2) = self.mod_fact(k, p);let (a3, e3) = self.mod_fact(n - k, p);if e1 > e2 + e3 {0} else {let moi = self.mod_inverse(a3 % p, p);(a1 * self.mod_inverse(a3 % p, p) % p) as i64}}}pub fn c(&mut self, n: usize, k: usize) -> i64 {let p = MOD as usize;if n == 0 && k == 0 {return 1;}if n == 0 {return 0;}if k == 0 {return 1;}if n < k {0} else {let (a1, e1) = self.mod_fact(n, p);let (a2, e2) = self.mod_fact(k, p);let (a3, e3) = self.mod_fact(n - k, p);if e1 > e2 + e3 {0} else {(((a1 * &self.moi[k]) % p * &self.moi[n - k]) % p) as i64}}}pub fn h(&mut self, n: usize, k: usize) -> i64 {return self.c(n + k - 1, k);}pub fn factorial(&mut self, n: usize) -> i64 {return self.p(n, n);}fn extgcd(&mut self, a: i64, b: i64) -> (i64, i64, i64) {if b == 0 {(a, 1, 0)} else {let (gcd, x, y) = extgcd(b, a % b);(gcd, y, x - (a / b) * y)}}fn mod_inverse(&mut self, a: usize, m: usize) -> usize {let (_, x, _) = self.extgcd(a as i64, m as i64);((m as i64 + x) as usize % m) % m}fn mod_fact(&mut self, n: usize, p: usize) -> (usize, usize) {if n == 0 {(1, 0)} else {let (a, b) = self.mod_fact(n / p, p);let pow = b + n / p;if n / p % 2 != 0 {(a * (p - self.f_table[(n % p) as usize]) % p, pow)} else {(a * self.f_table[(n % p) as usize] % p, pow)}}}}fn experiment(n: usize, m: usize) {let mut x = 1;for i in 0..m {x *= n;}let mut su = 0;for i in 2..=x {let d = divisors(i);let size = d.len();let a = i as f64;let b = (d[size / 2] + d[size / 2 - (size) % 2]) as f64 / 2.0;if b * b > a {// println!("{}", i);// d!(d);su += i;su %= UMOD;}}// d!(su);return;}fn solve_part() {let (n, m) = readuu();let x = M(n as i64).pow(m / 2);let x = x * (x + M(1)) * (x * M(2) + M(1)) / M(6);let y = M(n as i64).pow(m);// dbg!((x, y * (y + M(1)) / M(2)));let res = y * (y + M(1)) / M(2) - x;p!(res.0);// experiment(n, m);return;}fn solve() {let n: usize = read();for i in 0..n {solve_part();}return;}