leetcode1540. K 次操作转变字符串（Python3、c++）

leetcode1540. K 次操作转变字符串

给你两个字符串 s 和 t ，你的目标是在 k 次操作以内把字符串 s 转变成 t 。

在第 i 次操作时（1 <= i <= k），你可以选择进行如下操作：

• 选择字符串 s 中满足 1 <= j <= s.length 且之前未被选过的任意下标 j （下标从 1 开始），并将此位置的字符切换 i 次。
• 不进行任何操作。

切换 1 次字符的意思是用字母表中该字母的下一个字母替换它（字母表环状接起来，所以 'z' 切换后会变成 'a'）。

请记住任意一个下标 j 最多只能被操作 1 次。

如果在不超过 k 次操作内可以把字符串 s 转变成 t ，那么请你返回 true ，否则请你返回 false 。

示例 1：

输入：s = "input", t = "ouput", k = 9
输出：true
解释：第 6 次操作时，我们将 'i' 切换 6 次得到 'o' 。第 7 次操作时，我们将 'n' 切换 7 次得到 'u' 。

示例 2：

输入：s = "abc", t = "bcd", k = 10
输出：false
解释：我们需要将每个字符切换 1 次才能得到 t 。我们可以在第 1 次操作时将 'a' 切换成 'b' ，但另外 2 个字母在剩余操作中无法再转变为 t 中对应字母。

示例 3：

输入：s = "aab", t = "bbb", k = 27
输出：true
解释：第 1 次操作时，我们将第一个 'a' 切换 1 次得到 'b' 。在第 27 次操作时，我们将第二个字母 'a' 切换 27 次得到 'b' 。

提示：

• 1 <= s.length, t.length <= 10^5
• 0 <= k <= 10^9
• s 和 t 只包含小写英文字母。

---

方法：计数

思路：

本题使用计数的方法完成。对于每个需要改变的字符，它应该切换几次到达目标字符，取决于t[i]和s[i]之间的差，

• 如果t[i]对应字符在s[i]字符的后面，那么它需要的切换次数即为ord(t[i])-ord(s[i])
• 否则，需要切换的次数为ord(t[i])-ord(s[i])+26

因此，每个字符需要切换的次数均为0-25，因此我们使用一个长度为26的数组nums，保存需要切换i次的字符的个数。

因为我们只能在第i次操作的时候，切换i次，因此，如果nums[1] = 3，即有3个字符需要切换1次，那么我们只能在第1次操作、第27次操作和第53次操作将它们解决，即26(nums[i]-1)+i。*

因此我们对nums中的所有数进行遍历，根据上面公式，找到最大的值，跟k进行比较，返回。

还要注意的是，最开始如果s和t的长度不一样，直接返回False。

代码：

Python3：

class Solution:
    def canConvertString(self, s: str, t: str, k: int) -> bool:
        # 如果长度不一样，直接False
        n = len(s)
        m = len(t)
        if n != m:
            return False
        # 初始化nums
        nums = [0]*26
        # 遍历填写nums
        for i in range(n):
            delta = ord(t[i])-ord(s[i])
            if delta >= 0:
                nums[delta] += 1
            else:
                nums[delta+26] += 1
        # 初始化need，需要的次数
        need = 0
        # 开始更新need
        for i in range(1,26):
            temp = max(0,26*(nums[i]-1)+i)
            need = max(need,temp)
        return need <= k
        

cpp：

class Solution {
public:
    bool canConvertString(string s, string t, int k) {
         // 如果长度不一样，直接False
        int n = s.size();
        int m = t.size();
        if (n != m)
            return false;
        // 初始化nums
        vector<int>nums(26,0);
        // 遍历填写nums
        for (int i = 0; i < n; ++i){
            int delta = t[i]- s[i];
            if (delta >= 0)
                nums[delta] ++;
            else
                nums[delta+26] ++;
        }
            
        // 初始化need，需要的次数
        int need = 0;
        // 开始更新need
        for (int i = 1; i < 26; ++i){
            int temp = max(0,26*(nums[i]-1)+i);
            need = max(need,temp);
        }            
        return need <= k;
    }
};

结果：