PAT 乙级 1048 数字加密 (20分)

1048 数字加密 (20分)

本题要求实现一种数字加密方法。首先固定一个加密用正整数 A，对任一正整数 B，将其每 1 位数字与 A 的对应位置上的数字进行以下运算：对奇数位，对应位的数字相加后对 13 取余——这里用 J 代表 10、Q 代表 11、K 代表 12；对偶数位，用 B 的数字减去 A 的数字，若结果为负数，则再加 10。这里令个位为第 1 位。
输入格式：

输入在一行中依次给出 A 和 B，均为不超过 100 位的正整数，其间以空格分隔。
输出格式：

在一行中输出加密后的结果。
输入样例：

1234567 368782971
输出样例：

3695Q8118
作者: CHEN, Yue
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB

测试点3不过是因为在偶数位相减为0的时候，不需要加10
测试点4和测试点5不过是因为没考虑奇数位相加取余后为0，这时取值应为0

#include <iostream>
#include <algorithm>
using namespace std;
int main(){
    string a,b,ans;
    int num_a,num_b;
    char odd[13]={'0','1','2','3','4','5','6','7','8','9','J','Q','K'};
    ans = "";
    cin>>a>>b;
    if(a.length() > b.length()){
        int length = a.length() - b.length();
        for (int i = 0; i < length; ++i)
            b = "0" + b;
    }
    if(a.length() < b.length()){
        int length = b.length() - a.length();
        for (int i = 0; i < length; ++i)
            a = "0" + a;
    }
    reverse(a.begin(),a.end());
    reverse(b.begin(),b.end());
    for (int i = 0; i < a.length(); ++i) {
        num_a = int(a[i])-48;
        num_b = int(b[i])-48;
        if((i+1)%2==1){
            ans += odd[((num_a+num_b)%13)];
        } else{
            if(num_b - num_a >= 0)
                ans += char(num_b - num_a + 48);
            else
                ans += char(num_b - num_a + 10 + 48);
        }
    }
    reverse(ans.begin(),ans.end());
    cout<<ans<<endl;
    return 0;
}