蓝桥杯 ADV-225 算法提高 9-2 文本加密

问题描述
先编写函数EncryptChar,按照下述规则将给定的字符c转化（加密）为新的字符："A"转化"B"，"B"转化为"C"，... ..."Z"转化为"a"，"a"转化为"b",... ..., "z"转化为"A"，其它字符不加密。编写程序，加密给定字符串。
样例输出
与上面的样例输入对应的输出。
例：

数据规模和约定
输入数据中每一个数的范围。
例：50个字符以内无空格字符串。

#include <iostream>
#include <algorithm>
#include <vector>
#include <cctype>
#include <map>
using namespace std;
string encrypt(string s) {
    for (int i = 0; i < s.length(); i++) {
        if (s[i] == 'Z') s[i] = 'a';
        else if (s[i] == 'z') s[i] = 'A';
        else if (isalpha(s[i])) s[i]++;
    }
    return s;
}
int main() {
    string s;
    cin >> s;
    cout << encrypt(s);
    return 0;
}