No.482 密钥格式化

有一个密钥字符串 S ，只包含字母，数字以及 '-'（破折号）。其中， N 个 '-' 将字符串分成了 N+1 组。

给你一个数字 K，请你重新格式化字符串，使每个分组恰好包含 K 个字符。特别地，第一个分组包含的字符个数必须小于等于 K，但至少要包含 1 个字符。两个分组之间需要用 '-'（破折号）隔开，并且将所有的小写字母转换为大写字母。

给定非空字符串 S 和数字 K，按照上面描述的规则进行格式化。

示例 1：

输入：S = "5F3Z-2e-9-w", K = 4
输出："5F3Z-2E9W"
解释：字符串 S 被分成了两个部分，每部分 4 个字符；
     注意，两个额外的破折号需要删掉。

示例 2：

输入：S = "2-5g-3-J", K = 2
输出："2-5G-3J"
解释：字符串 S 被分成了 3 个部分，按照前面的规则描述，第一部分的字符可以少于给定的数量，其余部分皆为 2 个字符。

提示:

1. S 的长度可能很长，请按需分配大小。K 为正整数。
2. S 只包含字母数字（a-z，A-Z，0-9）以及破折号'-'
3. S 非空

思路分析

根据题目的思路可知，原字符串中的下划线对本题没什么用，所以统计的时候去掉。

首先计算每个部分可以分到多少个字符，也就是parts变量。然后计算第一个区块中有多少个字符，也就是remains。如果可以整除，第一个区块中就有k个字符，如果不能整除，第一个区块中的个数会小于k个，也就是valid(合法字符) / k。

在计算完成后，就可以将原字符串中的字符填充进去。注意：在填充完成remains个之后就填入一个下划线，并且将remains重置成k，因为往后的每一个区块之内都有k个字符。

Rust代码

#![allow(unused)]
fn main() {
struct Solution;
impl Solution {
    pub fn license_key_formatting(s: String, k: i32) -> String {
        let k = k as usize;
        let n = s.len();
        let count = s.chars().fold(0, |acc, x| acc + (x == '-') as usize);
        let valid = n - count;
        let parts = valid / k + (valid % k > 0) as usize;
        let mut remains = if valid % k == 0 {k} else {valid % k};
        let mut res = "".to_string();
        for ch in s.chars() {
            if ch == '-' {
                continue;
            }
            if remains == 0 {
                res.push('-');
                remains = k;
            }
            res.push(ch.to_ascii_uppercase());
            remains -= 1;
        }
        res
    }
}
}