20250318 finished.
This commit is contained in:
parent
8285dd6654
commit
c0a526d745
|
@ -544,3 +544,5 @@ mod p3110_score_of_a_string;
|
||||||
mod p2272_substring_with_largest_variance;
|
mod p2272_substring_with_largest_variance;
|
||||||
|
|
||||||
mod p1963_minimum_number_of_swaps_to_make_the_string_balanced;
|
mod p1963_minimum_number_of_swaps_to_make_the_string_balanced;
|
||||||
|
|
||||||
|
mod p2614_prime_in_diagonal;
|
||||||
|
|
52
src/problem/p2614_prime_in_diagonal.rs
Normal file
52
src/problem/p2614_prime_in_diagonal.rs
Normal file
|
@ -0,0 +1,52 @@
|
||||||
|
/**
|
||||||
|
* [2614] Prime In Diagonal
|
||||||
|
*/
|
||||||
|
pub struct Solution {}
|
||||||
|
|
||||||
|
// submission codes start here
|
||||||
|
|
||||||
|
impl Solution {
|
||||||
|
pub fn diagonal_prime(nums: Vec<Vec<i32>>) -> i32 {
|
||||||
|
let n = nums.len();
|
||||||
|
let mut result = 0;
|
||||||
|
|
||||||
|
for i in 0..n {
|
||||||
|
if Self::is_prime(nums[i][i]) {
|
||||||
|
result = result.max(nums[i][i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if Self::is_prime(nums[i][n - i - 1]) {
|
||||||
|
result = result.max(nums[i][n - i - 1]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
fn is_prime(num: i32) -> bool {
|
||||||
|
if num == 1 {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
!(2..=num.isqrt()).any(|x| num % x == 0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// submission codes end
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_2614() {
|
||||||
|
assert_eq!(
|
||||||
|
11,
|
||||||
|
Solution::diagonal_prime(vec![vec![1, 2, 3], vec![5, 6, 7], vec![9, 10, 11]])
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
17,
|
||||||
|
Solution::diagonal_prime(vec![vec![1, 2, 3], vec![5, 17, 7], vec![9, 10, 11]])
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user