PAT乙级真题 1048 数字加密 C++实现 (测试点2、5有坑)

题目

本题要求实现一种数字加密方法。首先固定一个加密用正整数 A，对任一正整数 B，将其每 1 位数字与 A 的对应位置上的数字进行以下运算：对奇数位，对应位的数字相加后对 13 取余——这里用 J 代表 10、Q 代表 11、K 代表 12；对偶数位，用 B 的数字减去 A 的数字，若结果为负数，则再加 10。这里令个位为第 1 位。
输入格式：
输入在一行中依次给出 A 和 B，均为不超过 100 位的正整数，其间以空格分隔。
输出格式：
在一行中输出加密后的结果。
输入样例：
1234567 368782971
输出样例：
3695Q8118

思路

用字符串记录两个数字，用i、j分别指向它们的末尾，同时向前移动。

用bool型的flag标记奇偶位，一开始为true，每移动一次改变一下，反转bool的值用“!”运算符：

flag = !flag

若是奇数位，需数字相加模13，即：

b[j] = (a[i] - 48 + b[j] - 48) % 13 + 48;

化简后为：

b[j] = (a[i] + b[j] - 5) % 13 + 48;

坑

一开始测试点2、5过不了，查了网上代码发现：

a的位数大于b时，b不足的位需要补0做运算！

其实当a位数小于b时，同样相当于a前面补0和b做运算，只不过b的值保持不变而已。

代码

#include <iostream>
using namespace std;

int main(){
    string a, b;
    cin >> a >> b;

    int i = a.length()-1;
    int j = b.length()-1;
    if (j<i){
        for (int k=0; k<i-j; k++){
            b = '0' + b;
        }
        j = i;
    }
    bool flag = true;
    while (i>=0 && j>=0){
        //奇数位
        if (flag){
            b[j] = (a[i] + b[j] - 5) % 13 + 48; 
            if (b[j]==58){
                b[j] = 'J';
            }
            else if (b[j]==59){
                b[j] = 'Q';
            }
            else if (b[j]==60){
                b[j] = 'K';
            }
        }
        else {
            b[j] -= a[i];
            if (b[j]<0){
                b[j] += 10;
            }
            b[j] += 48;
        }
        flag = !flag;
        i--;
        j--;
    }
    
    cout << b << endl;
    return 0;
}