20250513 finished.

This commit is contained in:
jackfiled 2025-05-13 16:02:00 +08:00
parent 9232c689c4
commit 7b7d6ca26d
2 changed files with 66 additions and 0 deletions

View File

@ -650,3 +650,5 @@ mod p2918_minimum_equal_sum_of_two_arrays_after_replacing_zeros;
mod p1550_three_consecutive_odds;
mod p2094_finding_3_digit_even_numbers;
mod p3335_total_characters_in_string_after_transformations_i;

View File

@ -0,0 +1,64 @@
/**
* [3335] Total Characters in String After Transformations I
*/
pub struct Solution {}
// submission codes start here
const MOD: i32 = 1_000_000_007;
impl Solution {
pub fn length_after_transformations(s: String, t: i32) -> i32 {
let mut map = vec![0; 26];
for c in s.bytes() {
map[(c - b'a') as usize] += 1;
}
let mut result = s.bytes().len() as i32;
for _ in 0..t {
let z_count = map[25];
for c in (0..25).rev() {
if map[c] != 0 {
map[c + 1] = (map[c + 1] + map[c]) % MOD;
map[c] = 0;
}
}
if z_count != 0 {
map[0] = (map[0] + z_count) % MOD;
map[1] = (map[1] + z_count) % MOD;
result = (result + z_count) % MOD;
map[25] = (map[25] + MOD - z_count) % MOD;
}
}
result
}
}
// submission codes end
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_3335() {
assert_eq!(
7,
Solution::length_after_transformations("abcyy".to_string(), 2)
);
assert_eq!(
5,
Solution::length_after_transformations("azbk".to_string(), 1)
);
assert_eq!(
79033769,
Solution::length_after_transformations("jqktcurgdvlibczdsvnsg".to_string(), 7517)
);
}
}