【PAT甲级题解记录】1023 Have Fun with Numbers (20 分)

【PAT甲级题解记录】1023 Have Fun with Numbers (20 分)

前言

Problem：1023 Have Fun with Numbers (20 分)

Tags：字符串

Difficulty：剧情模式 想流点汗 想流点血 死而无憾

Address：1023 Have Fun with Numbers (20 分)

问题描述

简单的字符串处理

参考代码

#include<iostream>
#include<vector>

using namespace std;

void solution_1023() {
    string s;
    cin >> s;
    vector<int> a;
    vector<int> num1(10, 0), num2(10, 0);

    // init
    for (int i = int(s.size() - 1); i >= 0; --i) {
        a.push_back(s[i] - '0');
        num1[s[i] - '0']++;
    }

    // double
    int add = 0;
    for (int i = 0; i < int(a.size()); ++i) {
        a[i] = a[i] * 2 + add;
        if (a[i] >= 10) {
            a[i] -= 10;
            add = 1;
        } else {
            add = 0;
        }
        num2[a[i]]++;
    }
    if (add) {
        a.push_back(1);
        num2[1]++;
    }

    // check
    for (int i = 0; i < 10; i++) {
        if (num1[i] != num2[i]) {
            cout << "No" << endl;
            for (int j = int(a.size() - 1); j >= 0; --j) {
                cout << a[j];
            }
            cout << endl;
            return;
        }
    }
    cout << "Yes" << endl;
    for (int j = int(a.size() - 1); j >= 0; --j) {
        cout << a[j];
    }
    cout << endl;
}
int main() {
    solution_1023();
    return 0;
}