20250318 finished.

This commit is contained in:
jackfiled 2025-03-18 13:38:56 +08:00
parent 8285dd6654
commit c0a526d745
2 changed files with 54 additions and 0 deletions

View File

@ -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;

View 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]])
);
}
}