leetcode1540. K 次操作转变字符串(Python3、c++)

文章目录

  • leetcode1540. K 次操作转变字符串
    • 方法:计数
      • 思路:
      • 代码:
        • Python3:
        • cpp:
      • 结果:

leetcode1540. K 次操作转变字符串

给你两个字符串 st ,你的目标是在 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
  • st 只包含小写英文字母。

方法:计数

思路:

本题使用计数的方法完成。对于每个需要改变的字符,它应该切换几次到达目标字符,取决于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;
    }
};

结果:

leetcode1540. K 次操作转变字符串(Python3、c++)_第1张图片

leetcode1540. K 次操作转变字符串(Python3、c++)_第2张图片

你可能感兴趣的:(Leetcode做题记录)