青训/简单：字符串解码问题

青训/简单：字符串解码问题

问题描述

问题描述
小R正在处理一个包含小写字母的字符串解码问题。给定一个长度为N的字符串S，其中包含小写英文字母。字符串的解码规则如下：

字符 'x' 应解码为 'y'，字符 'y' 应解码为 'x'。
字符 'a' 应解码为 'b'，字符 'b' 应解码为 'a'。
所有其他字符保持不变。
你的任务是返回解码后的字符串。

实例

def solution(N: int, S: str) -> str:
    # write code here
    pass

if __name__ == '__main__':
    print(solution(N = 5, S = "xaytq") == 'ybxtq')
    print(solution(N = 6, S = "abcxyz") == 'bacyxz')
    print(solution(N = 3, S = "zzz") == 'zzz')

思路

创建新字符串，循环原始字符

循环添加原始字符至新字符串，遇到对应两个匹配规则则替换更新字符串最后一位

答案

def solution(N: int, S: str) -> str:
    # 创建映射字典存储替换规则
    replace_map = {
        'x': 'y',
        'y': 'x',
        'a': 'b',
        'b': 'a'
    }
    
    # 将原始字符串转换为列表以便修改
    result = ''
    
    # 遍历字符串中的每个字符
    for c in S:
        # 如果字符在替换规则中，使用映射的字符
        # 否则保持原字符不变
        result += replace_map.get(c, c)
        
    return result

if __name__ == '__main__':
    print(solution(N = 5, S = "xaytq") == 'ybxtq')
    print(solution(N = 6, S = "abcxyz") == 'bacyxz')
    print(solution(N = 3, S = "zzz") == 'zzz')