PTA 7-2 电文环形解密

7-2 电文环形解密 (10分)

将26个英文字母按顺时针方向排成一个圆环。密钥是一个整数。

加密的方法：

若密钥 key≥0，则从明文字母开始按顺时针方向走 ∣key∣ 步得到密文字母；
若密钥 key<0，则从明文字母开始按逆时针方向走 ∣key∣ 步得到密文字母。
请编写解密程序，输入密钥(任意整数)和密文电文(含大、小写字母，数字和其他符号)，对电文中的大、小写字母进行解密，并输出明文电文。

输入格式
密钥(任意整数)
密文电文(以回车结束)

输出格式
明文电文(以回车结束)

输入样例1

2
K nqxg aqw!

输出样例1

I love you!

输入样例2

-2
G fytc 10 zyeq.

输出样例2

I have 10 bags.

代码：

#include <bits/stdc++.h>
using namespace std;
int main() {
    int a;
    cin>>a;
    string s;
    getchar();
    getline(cin,s);
    for (char & i : s) {
        if(i>='a'&&i<='z')
        {
            i=(26+i-'a'-a%26)%26+'a';
            //前边加一个26是防止逆时针超过‘a’
            //减去a%26而不是a是因为后三个测试点会很坏的输入很大个的a
        }
        else if(i>='A'&&i<='Z')
        {
            i=(26+i-'A'-a%26)%26+'A';
        }
    }
    cout<<s;
    return 0;
}